Fix to avoid fail when only import paddle

taichi-dev · Apr 29, 2022 · 9662f48 · 9662f48
1 parent d7cc03c
commit 9662f48
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/taichi/lang/kernel_impl.py b/python/taichi/lang/kernel_impl.py
@@ -658,7 +658,8 @@ def func__(*args):
                         ndarray_type.NdarrayType) and (self.match_ext_arr(v)):
                     has_external_arrays = True
                     is_numpy = isinstance(v, np.ndarray)
-                    is_torch = isinstance(v, torch.Tensor)
+                    is_torch = isinstance(v,
+                                          torch.Tensor) if has_torch else False
                     if is_numpy:
                         tmp = np.ascontiguousarray(v)
                         # Purpose: DO NOT GC |tmp|!