Arm64 use python

pytorch · Jul 5, 2023 · 38a91d1 · 38a91d1
1 parent 7d39a93
commit 38a91d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py
@@ -214,7 +214,7 @@ def smoke_test_modules():
                 subprocess.check_output(f"git clone --depth 1 {module['repo']}", stderr=subprocess.STDOUT, shell=True)
             try:
                 smoke_test_command = f"python3 {module['smoke_test']}"
-                if target_os == 'windows':
+                if target_os == 'windows' or target_os == 'macos-arm64':
                     smoke_test_command = f"python {module['smoke_test']}"
                 output = subprocess.check_output(
                     smoke_test_command, stderr=subprocess.STDOUT, shell=True,