From db7ee81c4e5593994b90e594921f3ee354709054 Mon Sep 17 00:00:00 2001 From: cchung100m Date: Sun, 2 Nov 2025 17:24:11 +0800 Subject: [PATCH] [#18397] Fix bug: Unsupported numpy or ml_dtypes dtype('O') when importing ONNX model using Relax frontend --- python/tvm/relax/frontend/onnx/onnx_frontend.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/tvm/relax/frontend/onnx/onnx_frontend.py b/python/tvm/relax/frontend/onnx/onnx_frontend.py index 3b94ba1d6672..2e4e7a3125e9 100644 --- a/python/tvm/relax/frontend/onnx/onnx_frontend.py +++ b/python/tvm/relax/frontend/onnx/onnx_frontend.py @@ -340,6 +340,8 @@ def base_impl(cls, bb, inputs, attr, params): x = _to_numpy(inputs[0]) y = _to_numpy(inputs[1]) output = cls.numpy_op(x, y) # pylint: disable=not-callable + if isinstance(x, relax.PrimValue) and isinstance(y, relax.PrimValue): + return relax.PrimValue(output.item()) if x.dtype == y.dtype: # no numpy precision widening output = output.astype(x.dtype)