Skip to content

Commit e0cc7d4

Browse files
committed
ENH: read dtype properly for complex dtypes
1 parent 3d1181a commit e0cc7d4

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

kokkos/utility.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -99,8 +99,10 @@ def read_dtype(_dtype):
9999
return lib.uint64
100100
elif _dtype == np.float32:
101101
return lib.float32
102-
elif _dtype == np.float64:
103-
return lib.float64
102+
elif _dtype == np.complex64:
103+
return lib.complex_float32_dtype
104+
elif _dtype == np.complex128:
105+
return lib.complex_float64_dtype
104106
except ImportError:
105107
pass
106108

0 commit comments

Comments
 (0)