From 005868279ba8ce5f9208c5293115b1d626497016 Mon Sep 17 00:00:00 2001 From: atalman Date: Thu, 16 Nov 2023 07:07:55 -0800 Subject: [PATCH 1/2] test --- test/smoke_test/smoke_test.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py index c0bb9230f..76f41b685 100644 --- a/test/smoke_test/smoke_test.py +++ b/test/smoke_test/smoke_test.py @@ -9,6 +9,7 @@ import torch._dynamo import torch.nn as nn import torch.nn.functional as F +from pathlib import Path gpu_arch_ver = os.getenv("MATRIX_GPU_ARCH_VERSION") gpu_arch_type = os.getenv("MATRIX_GPU_ARCH_TYPE") From 618b326b72cca410e2c8b1166b491f74e4c2ad85 Mon Sep 17 00:00:00 2001 From: atalman Date: Thu, 16 Nov 2023 07:15:01 -0800 Subject: [PATCH 2/2] basedir --- test/smoke_test/smoke_test.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py index 76f41b685..a14ad4316 100644 --- a/test/smoke_test/smoke_test.py +++ b/test/smoke_test/smoke_test.py @@ -17,7 +17,7 @@ stable_version = os.getenv("MATRIX_STABLE_VERSION") package_type = os.getenv("MATRIX_PACKAGE_TYPE") target_os = os.getenv("TARGET_OS") -BASE_DIR = Path(__file__).parent.parent +BASE_DIR = Path(__file__).parent.parent.parent is_cuda_system = gpu_arch_type == "cuda" NIGHTLY_ALLOWED_DELTA = 3 @@ -85,6 +85,8 @@ def check_version(package: str) -> None: raise RuntimeError( f"{module['name']} version mismatch, expected {release_version[module['name']]} for channel {channel}. But its {module_version}" ) + else: + print(f"{module['name']} version actual: {module_version} expected: {release_version[module['name']]} for channel {channel}.") else: print(f"Skip version check for channel {channel} as stable version is None")