[Fixbug] Convert tensor to cpu before dumping (#141)

convert tensor to cpu before dumping
hidet-org · Mar 23, 2023 · 3f15236 · 3f15236
1 parent 7fbb885
commit 3f15236
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/hidet/graph/tensor.py b/python/hidet/graph/tensor.py
@@ -422,7 +422,7 @@ def __hash__(self):
 
     def __getstate__(self):
         if self.storage:
-            data = self.detach().numpy()
+            data = self.detach().cpu().numpy()
         else:
             data = None