From c2a761dde577925c5d9b00bcd72c41c39eac8624 Mon Sep 17 00:00:00 2001 From: Ruihang Lai Date: Sun, 24 Mar 2024 15:54:45 -0400 Subject: [PATCH] [Fix] Fix numpy dtype map NumPy 2.0 removes the dtype `np.float_`, which may introduces compatibility issue for TVM. --- python/tvm/_ffi/runtime_ctypes.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/tvm/_ffi/runtime_ctypes.py b/python/tvm/_ffi/runtime_ctypes.py index 570a24ed5dd3..fd9f4beb4374 100644 --- a/python/tvm/_ffi/runtime_ctypes.py +++ b/python/tvm/_ffi/runtime_ctypes.py @@ -95,8 +95,9 @@ class DataType(ctypes.Structure): np.dtype(np.float16): "float16", np.dtype(np.float32): "float32", np.dtype(np.float64): "float64", - np.dtype(np.float_): "float64", } + if np.__version__.startswith("1."): + NUMPY2STR[np.dtype(np.float_)] = "float64" STR2DTYPE = { "void": {"type_code": DataTypeCode.HANDLE, "bits": 0, "lanes": 0}, "bool": {"type_code": DataTypeCode.UINT, "bits": 1, "lanes": 1},