Skip to content

Commit 9ebc74c

Browse files
committed
fix
1 parent 4092066 commit 9ebc74c

4 files changed

Lines changed: 7 additions & 7 deletions

File tree

_scripts/test_backend_onnxruntime.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ def run_node(cls, node, inputs, device=None, outputs_info=None, **kwargs):
141141

142142
backend_test.exclude("(test_adagrad|test_adam|test_add_uint8)")
143143

144-
if pv.Version(onnxruntime.__version__) <= pv.Version("1.25"):
144+
if pv.Version(onnxruntime.__version__) <= pv.Version("1.30"):
145145
backend_test.exclude("(test_attention_4d_with|test_attention_4d_gqa)")
146146

147147

_unittests/ut_reference/test_backend_onnxruntime_evaluator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -299,7 +299,7 @@ def run_node(cls, node, inputs, device=None, outputs_info=None, **kwargs):
299299
)
300300

301301

302-
if pv.Version(onnxruntime.__version__) <= pv.Version("1.25"):
302+
if pv.Version(onnxruntime.__version__) <= pv.Version("1.30"):
303303
backend_test.exclude("(test_attention_4d_with|test_attention_4d_gqa)")
304304

305305
# import all test cases at global scope to make them visible to python.unittest

_unittests/ut_torch_export_patches/test_patch_transformers.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -645,7 +645,7 @@ def test_plug_multi_head_attention_qwen25_packed_float16(self):
645645
self.assertEqualArray(results.eager_outputs[0], results.onnx_outputs[0], atol=0.01)
646646
self.assertLess(results.diffs[0]["abs"], 0.01)
647647

648-
@requires_onnxruntime("1.25")
648+
@requires_onnxruntime("1.30")
649649
@unittest.skipIf(not patch_qwen2_5, "Qwen25 not part of this transformers")
650650
def test_plug_multi_head_attention_qwen25_loopmha_float16(self):
651651
from onnx_diagnostic.torch_export_patches.patches._patch_transformers_qwen2_5 import (
@@ -681,7 +681,7 @@ def test_plug_multi_head_attention_qwen25_loopmha_float16(self):
681681
self.assertEqualArray(results.eager_outputs[0], results.onnx_outputs[0], atol=0.01)
682682
self.assertLess(results.diffs[0]["abs"], 0.01)
683683

684-
@requires_onnxruntime("1.25")
684+
@requires_onnxruntime("1.30")
685685
@unittest.skipIf(not patch_qwen2_5, "Qwen25 not part of this transformers")
686686
def test_plug_multi_head_attention_qwen25_loopmha_float32(self):
687687
from onnx_diagnostic.torch_export_patches.patches._patch_transformers_qwen2_5 import (
@@ -717,7 +717,7 @@ def test_plug_multi_head_attention_qwen25_loopmha_float32(self):
717717
self.assertEqualArray(results.eager_outputs[0], results.onnx_outputs[0], atol=1e-5)
718718
self.assertLess(results.diffs[0]["abs"], 1e-5)
719719

720-
@requires_onnxruntime("1.25")
720+
@requires_onnxruntime("1.30")
721721
@unittest.skipIf(not patch_qwen2_5, "Qwen25 not part of this transformers")
722722
def test_plug_multi_head_attention_qwen25_loopa24_float16(self):
723723
from onnx_diagnostic.torch_export_patches.patches._patch_transformers_qwen2_5 import (
@@ -748,7 +748,7 @@ def test_plug_multi_head_attention_qwen25_loopa24_float16(self):
748748
)
749749
self.assertLess(results.diffs[0]["abs"], 0.005)
750750

751-
@requires_onnxruntime("1.25")
751+
@requires_onnxruntime("1.30")
752752
@unittest.skipIf(not patch_qwen2_5, "Qwen25 not part of this transformers")
753753
def test_plug_multi_head_attention_qwen25_loopa24_float32(self):
754754
from onnx_diagnostic.torch_export_patches.patches._patch_transformers_qwen2_5 import (

_unittests/ut_torch_onnx/test_discrepancies.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def qwen_sdpa_attention(
4646
return attn_output
4747

4848
for model_name in ["attention_loopa24.onnx", "attention_loopmha.onnx"]:
49-
if model_name == "attention_loopa24.onnx" and not has_onnxruntime("1.25"):
49+
if model_name == "attention_loopa24.onnx" and not has_onnxruntime("1.30"):
5050
# not available
5151
continue
5252
with self.subTest(model=model_name):

0 commit comments

Comments
 (0)