fix mass_spring_3d_ggui backend (#5127)

taichi-dev · Jun 12, 2022 · 573f89e · 573f89e
1 parent e10a11b
commit 573f89e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/taichi/examples/ggui_examples/mass_spring_3d_ggui.py b/python/taichi/examples/ggui_examples/mass_spring_3d_ggui.py
@@ -1,5 +1,7 @@
 import taichi as ti
-ti.init(arch=ti.gpu)  # Alternatively, ti.init(arch=ti.cpu)
+
+arch = ti.vulkan if ti._lib.core.with_vulkan() else ti.cuda
+ti.init(arch=arch)
 
 n = 128
 quad_size = 1.0 / n