separate tests for convert pytorch tensor

huggingface · Oct 25, 2023 · cd56f6c · cd56f6c
1 parent 1da21cd
commit cd56f6c
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 13 deletions.
diff --git a/candle-pyo3/test.py b/candle-pyo3/test.py
@@ -1,5 +1,4 @@
 import candle
-import torch
 
 print(f"mkl:         {candle.utils.has_mkl()}")
 print(f"accelerate:  {candle.utils.has_accelerate()}")
@@ -30,15 +29,3 @@
 dequant_t = quant_t.dequantize()
 diff2 = (t - dequant_t).sqr()
 print(diff2.mean_all())
-
-# convert from candle tensor to torch tensor
-t = candle.randn((3, 512, 512))
-torch_tensor = t.to_torch()
-print(torch_tensor)
-print(type(torch_tensor))
-
-# convert from torch tensor to candle tensor
-t = torch.randn((3, 512, 512))
-candle_tensor = candle.Tensor(t)
-print(candle_tensor)
-print(type(candle_tensor))
diff --git a/candle-pyo3/test_pytorch.py b/candle-pyo3/test_pytorch.py
@@ -0,0 +1,14 @@
+import candle
+import torch
+
+# convert from candle tensor to torch tensor
+t = candle.randn((3, 512, 512))
+torch_tensor = t.to_torch()
+print(torch_tensor)
+print(type(torch_tensor))
+
+# convert from torch tensor to candle tensor
+t = torch.randn((3, 512, 512))
+candle_tensor = candle.Tensor(t)
+print(candle_tensor)
+print(type(candle_tensor))