From a1305951f669f2a57231fee88db6497601010b76 Mon Sep 17 00:00:00 2001 From: Xuan Xie Date: Sat, 7 Mar 2026 14:55:53 -0800 Subject: [PATCH] fix get_converter bug --- python/tvm/relax/frontend/onnx/onnx_frontend.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/python/tvm/relax/frontend/onnx/onnx_frontend.py b/python/tvm/relax/frontend/onnx/onnx_frontend.py index 3dc575ae778c..ba1058225a1f 100644 --- a/python/tvm/relax/frontend/onnx/onnx_frontend.py +++ b/python/tvm/relax/frontend/onnx/onnx_frontend.py @@ -303,7 +303,10 @@ def get_converter(cls, opset): """ versions = [int(d.replace("_impl_v", "")) for d in dir(cls) if "_impl_v" in d] versions = sorted(versions + [opset]) - version = versions[max([i for i, v in enumerate(versions) if v == opset]) - 1] + opset_idx = max([i for i, v in enumerate(versions) if v == opset]) + if opset_idx == 0: + raise NotImplementedError(f"opset version {opset} of {cls.__name__} not implemented") + version = versions[opset_idx - 1] if hasattr(cls, f"_impl_v{version}"): return getattr(cls, f"_impl_v{version}") raise NotImplementedError(f"opset version {version} of {cls.__name__} not implemented")