Skip to content

Commit d89735e

Browse files
committed
changes in onnx
1 parent 95982d1 commit d89735e

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

_scripts/test_backend_onnxruntime.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
import unittest
66
import warnings
77
from typing import Any
8+
import packaging.version as pv
89
import numpy
910
import onnx.backend.base
1011
import onnx.backend.test
@@ -140,6 +141,12 @@ def run_node(cls, node, inputs, device=None, outputs_info=None, **kwargs):
140141

141142
backend_test.exclude("(test_adagrad|test_adam|test_add_uint8)")
142143

144+
if pv.Version(onnxruntime.__version__) < pv.Version("1.24"):
145+
backend_test.exclude(
146+
"(test_attention_4d_with_qk|test_attention_4d_gqa|"
147+
"test_attention_4d_with_past_and_present_qk)"
148+
)
149+
143150

144151
# import all test cases at global scope to make them visible to python.unittest
145152
globals().update(backend_test.test_cases)

0 commit comments

Comments
 (0)