saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-26 22:35:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Skip python arena shrinkage test on ppc (#10901)

Browse source
This commit is contained in:
Kotaro Yamamoto 2022-03-19 11:31:21 +09:00 committed by GitHub
parent de06d95096
commit 2dea7dc27f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
onnxruntime/test/python/onnxruntime_test_python.py
View file

@ -1069,9 +1069,9 @@ class TestInferenceSession(unittest.TestCase):
onnxrt.InferenceSession(get_name("mul_1.onnx"), sess_options=so2, providers=onnxrt.get_available_providers())
def testMemoryArenaShrinkage(self):
if platform.architecture()[0] == '32bit':
# on x86 builds, the CPU allocator does not use an arena
print("Skipping testMemoryArenaShrinkage in 32bit platform.")
if platform.architecture()[0] == '32bit' or 'ppc' in platform.machine() or 'powerpc' in platform.machine():
# on x86 or ppc builds, the CPU allocator does not use an arena
print("Skipping testMemoryArenaShrinkage in 32bit or powerpc platform.")
else:
x = np.array([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]], dtype=np.float32)