3
3
import unittest
4
4
import numpy as np
5
5
import onnx
6
- from onnx_array_api .ext_test_case import ExtTestCase
6
+ from onnx_array_api .ext_test_case import ExtTestCase , skipif_ci_apple
7
7
from onnx_array_api .graph_api .graph_builder import GraphBuilder , OptimizationOptions
8
8
from onnx_array_api .reference import (
9
9
from_array_extended ,
@@ -107,6 +107,7 @@ def test_simple_big(self):
107
107
got = ref .run (None , feeds )
108
108
self .assertEqualArray (expected , got [0 ])
109
109
110
+ @skipif_ci_apple ("libomp is missing" )
110
111
def test_constant_folding (self ):
111
112
with contextlib .redirect_stdout (io .StringIO ()):
112
113
g = GraphBuilder (verbose = 10 )
@@ -133,6 +134,7 @@ def test_constant_folding(self):
133
134
got = ref .run (None , feeds )
134
135
self .assertEqualArray (expected , got [0 ])
135
136
137
+ @skipif_ci_apple ("libomp is missing" )
136
138
def test_constant_folding2 (self ):
137
139
g = GraphBuilder (
138
140
optimization_options = OptimizationOptions (constant_folding = True )
@@ -270,6 +272,7 @@ def test_remove_unused_nodes_simple(self):
270
272
got = ref .run (None , feeds )
271
273
self .assertEqualArray (expected , got [0 ])
272
274
275
+ @skipif_ci_apple ("libomp is missing" )
273
276
def test_constant_array (self ):
274
277
with contextlib .redirect_stdout (io .StringIO ()):
275
278
g = GraphBuilder (verbose = 10 )
@@ -290,6 +293,7 @@ def test_constant_array(self):
290
293
got = ref .run (None , feeds )
291
294
self .assertEqualArray (expected , got [0 ])
292
295
296
+ @skipif_ci_apple ("libomp is missing" )
293
297
def test_constant_array_2 (self ):
294
298
with contextlib .redirect_stdout (io .StringIO ()):
295
299
g = GraphBuilder (verbose = 10 )
0 commit comments