From 235983e9e7a5563594d18b13b752e58a08263b49 Mon Sep 17 00:00:00 2001 From: "Gao, Xiang" Date: Wed, 17 Dec 2025 11:12:28 -0800 Subject: [PATCH 1/2] Update error regex in for topk in opinfo_input_generators.py --- tests/python/opinfo/opinfo_input_generators.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/opinfo/opinfo_input_generators.py b/tests/python/opinfo/opinfo_input_generators.py index 1efd8dea6b8..c3374441e95 100644 --- a/tests/python/opinfo/opinfo_input_generators.py +++ b/tests/python/opinfo/opinfo_input_generators.py @@ -924,7 +924,7 @@ def topk_error_generator( # error coming from aten fallback. yield SampleInput( a, 16, 1, True, False - ), RuntimeError, "selected index k out of range" + ), RuntimeError, "k .* range" def index_select_generator( From 74e402f593569a2192d52770a93d8e00f09a99b3 Mon Sep 17 00:00:00 2001 From: "Gao, Xiang" Date: Wed, 17 Dec 2025 11:15:50 -0800 Subject: [PATCH 2/2] Update opinfo_input_generators.py --- tests/python/opinfo/opinfo_input_generators.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/python/opinfo/opinfo_input_generators.py b/tests/python/opinfo/opinfo_input_generators.py index c3374441e95..01c72fe2aef 100644 --- a/tests/python/opinfo/opinfo_input_generators.py +++ b/tests/python/opinfo/opinfo_input_generators.py @@ -922,9 +922,7 @@ def topk_error_generator( ), RuntimeError, "Invalid resized domain extent" # error coming from aten fallback. - yield SampleInput( - a, 16, 1, True, False - ), RuntimeError, "k .* range" + yield SampleInput(a, 16, 1, True, False), RuntimeError, "k .* range" def index_select_generator(