[example] Make implicit_fem fallback to CPU when CUDA is not available (

#5512)
taichi-dev · Jul 25, 2022 · b6a322a · b6a322a
1 parent 6c950c9
commit b6a322a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/taichi/examples/simulation/implicit_fem.py b/python/taichi/examples/simulation/implicit_fem.py
@@ -14,10 +14,10 @@
 parser.add_argument('place_holder', nargs='*')
 args = parser.parse_args()
 
-ti.init(arch=ti.gpu, dynamic_index=True)
+ti.init(arch=ti.cuda, dynamic_index=True)
 
 if args.gui == 'auto':
-    if _ti_core.GGUI_AVAILABLE:
+    if _ti_core.GGUI_AVAILABLE and ti.lang.impl.current_cfg().arch == ti.cuda:
         args.gui = 'ggui'
     else:
         args.gui = 'cpu'