diff --git a/.github/workflows/validate-linux-binaries.yml b/.github/workflows/validate-linux-binaries.yml index 0bb5e3ce7..5e1a00698 100644 --- a/.github/workflows/validate-linux-binaries.yml +++ b/.github/workflows/validate-linux-binaries.yml @@ -71,7 +71,7 @@ jobs: export TORCH_ONLY=${{ inputs.torchonly }} export TARGET_OS="linux" eval "$(conda shell.bash hook)" - printf '%s\n' ${{ needs.generate-release-matrix.outputs.matrix }} > release_matrix.json + printf '%s\n' ${{ toJson(needs.generate-release-matrix.outputs.matrix) }} > release_matrix.json # Special case PyPi installation package. And Install of PyPi package via poetry if [[ ${MATRIX_PACKAGE_TYPE} == "manywheel" && ${MATRIX_GPU_ARCH_VERSION} == "12.1" ]]; then diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py index c0bb9230f..a14ad4316 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") @@ -16,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 @@ -84,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")