From 2dea7dc27f8d4382a042a0e6487fc925d36223fc Mon Sep 17 00:00:00 2001 From: Kotaro Yamamoto Date: Sat, 19 Mar 2022 11:31:21 +0900 Subject: [PATCH] Skip python arena shrinkage test on ppc (#10901) --- onnxruntime/test/python/onnxruntime_test_python.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/onnxruntime/test/python/onnxruntime_test_python.py b/onnxruntime/test/python/onnxruntime_test_python.py index 6c9d0540a2..858f18c97e 100644 --- a/onnxruntime/test/python/onnxruntime_test_python.py +++ b/onnxruntime/test/python/onnxruntime_test_python.py @@ -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)