Skip to content

Commit

Permalink
Update implicit_mass_spring.py
Browse files Browse the repository at this point in the history
  • Loading branch information
jim19930609 authored Dec 8, 2022
1 parent 914533c commit a6490d5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions python/taichi/examples/simulation/implicit_mass_spring.py
Original file line number Diff line number Diff line change
Expand Up @@ -252,9 +252,9 @@ def main():
args, unknowns = parser.parse_known_args()
arch = args.arch
if arch in ["x64", "cpu", "arm64"]:
ti.init(arch=ti.cpu, real_matrix=True, real_matrix_scalarize=True)
ti.init(arch=ti.cpu)
elif arch in ["cuda", "gpu"]:
ti.init(arch=ti.cuda, real_matrix=True, real_matrix_scalarize=True)
ti.init(arch=ti.cuda)
else:
raise ValueError('Only CPU and CUDA backends are supported for now.')

Expand Down

0 comments on commit a6490d5

Please sign in to comment.