Update taichi/backends/vulkan/aot_module_builder_impl.cpp

taichi-dev · Feb 25, 2022 · 8fd0149 · 8fd0149
1 parent 512a6ba
commit 8fd0149
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/taichi/backends/vulkan/aot_module_builder_impl.cpp b/taichi/backends/vulkan/aot_module_builder_impl.cpp
@@ -54,7 +54,7 @@ class AotDataConverter {
         arr_arg.dtype_name = arg.dt.to_string();
         arr_arg.field_dim = arg.field_dim;
         arr_arg.element_shape = arg.element_shape;
-        arr_arg.shape_offset_in_args_buf = arg.index * sizeof(int);
+        arr_arg.shape_offset_in_args_buf = arg.index * sizeof(int32_t);
         res.arr_args[arg.index] = arr_arg;
       }
     }