[Bug] Fix the always-in-cpu from_numpy in ti.ndarray (taichi-dev#3239)

NoobTitan · Oct 21, 2021 · fdad691 · fdad691
1 parent fcd5a88
commit fdad691
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/taichi/lang/_ndarray.py b/python/taichi/lang/_ndarray.py
@@ -125,6 +125,8 @@ def from_numpy(self, arr):
             )
         if impl.current_cfg().ndarray_use_torch:
             self.arr = torch.from_numpy(arr).to(self.arr.dtype)
+            if impl.current_cfg().arch == _ti_core.Arch.cuda:
+                self.arr = self.arr.cuda()
         else:
             if hasattr(arr, 'contiguous'):
                 arr = arr.contiguous()