From a63733921e0942333014d75678ad8b702ceb98af Mon Sep 17 00:00:00 2001
From: Ranjit Ranjan <ranjit.sastra@gmail.com>
Date: Tue, 16 May 2023 03:20:41 -0500
Subject: [PATCH 7/7] testcase fixes

---
 cmake/onnxruntime_unittests.cmake            | 2 ++
 onnxruntime/python/onnxruntime_validation.py | 1 +
 2 files changed, 3 insertions(+)

diff --git a/cmake/onnxruntime_unittests.cmake b/cmake/onnxruntime_unittests.cmake
index cc8db8392..51572c01c 100644
--- a/cmake/onnxruntime_unittests.cmake
+++ b/cmake/onnxruntime_unittests.cmake
@@ -817,6 +817,8 @@ if (${CMAKE_SYSTEM_NAME} MATCHES "AIX")
   target_link_options(onnxruntime_test_all PRIVATE "-Wl,-bkeepfile:CMakeFiles/onnxruntime_test_all.dir${TEST_SRC_DIR}/quantization/quantization_test.cc.o")
   target_link_options(onnxruntime_test_all PRIVATE "-Wl,-bkeepfile:CMakeFiles/onnxruntime_test_all.dir${TEST_SRC_DIR}/framework/tensor_test.cc.o")
   target_link_options(onnxruntime_test_all PRIVATE "-Wl,-bkeepfile:CMakeFiles/onnxruntime_test_all.dir${TEST_SRC_DIR}/framework/bfc_arena_test.cc.o")
+  target_link_options(onnxruntime_test_all PRIVATE "-Wl,-bkeepfile:CMakeFiles/onnxruntime_test_all.dir${TEST_SRC_DIR}/platform/threadpool_test.cc.o")
+  target_link_options(onnxruntime_test_all PRIVATE "-Wl,-bkeepfile:CMakeFiles/onnxruntime_test_all.dir${TEST_SRC_DIR}/common/narrow_test.cc.o")
 endif()
 
 # the default logger tests conflict with the need to have an overall default logger
diff --git a/onnxruntime/python/onnxruntime_validation.py b/onnxruntime/python/onnxruntime_validation.py
index 53c54d4b0..ea147c801 100644
--- a/onnxruntime/python/onnxruntime_validation.py
+++ b/onnxruntime/python/onnxruntime_validation.py
@@ -56,6 +56,7 @@ def check_distro_info():
             warnings.warn('Unsupported macOS version (%s). ONNX Runtime supports macOS 11.0 or later.' %
                           (__my_distro_ver__))
     elif __my_system__ == "aix":
+        import subprocess
         __my_distro__ = __my_system__
         returned_output = subprocess.check_output("oslevel")
         __my_distro_ver__str = returned_output.decode("utf-8")
-- 
2.38.3

