Update smoke_test.py

juliagmt-google · Aug 19, 2024 · b6394eb · b6394eb
1 parent e18dd8f
commit b6394eb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py
@@ -15,6 +15,7 @@
     gpu_arch_ver = os.getenv("MATRIX_GPU_ARCH_VERSION")
 else:
     gpu_arch_ver = os.getenv("GPU_ARCH_VERSION")  # Use fallback if available
+gpu_arch_type = os.getenv("MATRIX_GPU_ARCH_TYPE")
 channel = os.getenv("MATRIX_CHANNEL")
 package_type = os.getenv("MATRIX_PACKAGE_TYPE")
 target_os = os.getenv("TARGET_OS", sys.platform)