[vulkan] Fix bug in empty root buffer (#2807)

* [vulkan] Fix bug in empty root buffer * Auto Format * typo Co-authored-by: Taichi Gardener <[email protected]>
taichi-dev · Aug 26, 2021 · d85729c · d85729c
1 parent 746129b
commit d85729c
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/taichi/backends/vulkan/spirv_snode_compiler.cpp b/taichi/backends/vulkan/spirv_snode_compiler.cpp
@@ -12,11 +12,15 @@ class SpirvSNodeCompiler {
   CompiledSpirvSNode run(IRBuilder *builder,
                          const CompiledSNodeStructs *compiled_structs) {
     CompiledSpirvSNode result;
-    result.root_stype = compute_snode_stype(
-        builder, compiled_structs,
-        compiled_structs->snode_descriptors.find(compiled_structs->root->id)
-            ->second,
-        &result.snode_id_struct_stype_tbl, &result.snode_id_array_stype_tbl);
+    if (compiled_structs->root_size != 0) {
+      result.root_stype = compute_snode_stype(
+          builder, compiled_structs,
+          compiled_structs->snode_descriptors.find(compiled_structs->root->id)
+              ->second,
+          &result.snode_id_struct_stype_tbl, &result.snode_id_array_stype_tbl);
+    } else {  // Use an arbitary default type to skip empty root buffer
+      result.root_stype = builder->i32_type();
+    }
     return result;
   }