From a88f42509513c60d6e7d3703fe566ca6dbdb3820 Mon Sep 17 00:00:00 2001 From: siahuat0727 Date: Tue, 8 Mar 2022 19:16:16 +0800 Subject: [PATCH] Fix mapping error (#390) Co-authored-by: Wenxiang Hu <8460860+wenxcs@users.noreply.github.com> --- src/python/nnfusion/dtypes.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/python/nnfusion/dtypes.py b/src/python/nnfusion/dtypes.py index b05a41bf3..e20ef0f45 100644 --- 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":