fix 1920

taichi-dev · Oct 10, 2020 · 9a8a53e · 9a8a53e
1 parent eba8cff
commit 9a8a53e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/taichi/lang/meta.py b/python/taichi/lang/meta.py
@@ -28,7 +28,7 @@ def vector_to_fast_image(img: ti.template(), out: ti.ext_arr()):
         u, v, w = min(255, max(0, int(img[i, img.shape[1] - 1 - j] * 255)))
         # We use i32 for |out| since OpenGL and Metal doesn't support u8 types
         # TODO: treat Cocoa and Big-endian machines, with XOR logic
-        out[j * 1920 + i] = w + (v << 8) + (u << 16)
+        out[j * img.shape[0] + i] = w + (v << 8) + (u << 16)
 
 
 @ti.kernel