Fix mapping error (#390)

Co-authored-by: Wenxiang Hu <[email protected]>
microsoft · Mar 8, 2022 · a88f425 · a88f425
1 parent 53929d5
commit a88f425
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/python/nnfusion/dtypes.py b/src/python/nnfusion/dtypes.py
@@ -20,8 +20,8 @@
     TypeObject._make(
         ["float64", ctypes.c_double, torch.float64, numpy.float64]),
     "float64":
-    TypeObject._make(["float64", ctypes.c_float, torch.float32,
-                      numpy.float32]),
+    TypeObject._make(["float64", ctypes.c_double, torch.float64,
+                      numpy.float64]),
     "int8":
     TypeObject._make(["int8", ctypes.c_int8, torch.int8, numpy.int8]),
     "int16":