diff --git a/build_tools/packaging/python/templates/rocm/src/rocm_sdk/tests/devel_test.py b/build_tools/packaging/python/templates/rocm/src/rocm_sdk/tests/devel_test.py index 3e1e2619b48..d7618e863e9 100644 --- a/build_tools/packaging/python/templates/rocm/src/rocm_sdk/tests/devel_test.py +++ b/build_tools/packaging/python/templates/rocm/src/rocm_sdk/tests/devel_test.py @@ -135,7 +135,7 @@ def testSharedLibrariesLoad(self): extra_setup = "" if ( "hipdnn_plugins" in str(so_path) or "test_plugins" in str(so_path) - ) and sys.platform == "win32": + ) and platform.system() == "Windows": # hipdnn plugins have dependencies on other libraries (e.g. miopen). # In a real-world scenario, hipdnn_backend loads these plugins, and # the dependencies are found because they reside in the same directory diff --git a/build_tools/packaging/python/templates/rocm/src/rocm_sdk/tests/libraries_test.py b/build_tools/packaging/python/templates/rocm/src/rocm_sdk/tests/libraries_test.py index fec5cbe1601..db8778e439b 100644 --- a/build_tools/packaging/python/templates/rocm/src/rocm_sdk/tests/libraries_test.py +++ b/build_tools/packaging/python/templates/rocm/src/rocm_sdk/tests/libraries_test.py @@ -5,6 +5,7 @@ import importlib import os from pathlib import Path +import platform import subprocess import sys import sysconfig @@ -63,7 +64,9 @@ def testSharedLibrariesLoad(self): continue extra_setup = "" - if "hipdnn_plugins" in str(so_path) and sys.platform == "win32": + if ( + "hipdnn_plugins" in str(so_path) or "test_plugins" in str(so_path) + ) and platform.system() == "Windows": # hipdnn plugins have dependencies on other libraries (e.g. miopen). # In a real-world scenario, hipdnn_backend loads these plugins, and # the dependencies are found because they reside in the same directory