diff --git a/.github/workflows/test_build_wheels_linux_aarch64_without_cuda.yml b/.github/workflows/test_build_wheels_linux_aarch64_without_cuda.yml index f354e5551f..d5a2565a0f 100644 --- a/.github/workflows/test_build_wheels_linux_aarch64_without_cuda.yml +++ b/.github/workflows/test_build_wheels_linux_aarch64_without_cuda.yml @@ -9,6 +9,7 @@ on: - .github/workflows/generate_binary_build_matrix.yml - .github/workflows/test_build_wheels_linux_aarch64_without_cuda.yml - tools/scripts/generate_binary_build_matrix.py + - tools/pkg-helpers/pytorch_pkg_helpers/version.py workflow_dispatch: permissions: diff --git a/.github/workflows/test_build_wheels_linux_without_cuda.yml b/.github/workflows/test_build_wheels_linux_without_cuda.yml index 0dee2568be..989f8a82c5 100644 --- a/.github/workflows/test_build_wheels_linux_without_cuda.yml +++ b/.github/workflows/test_build_wheels_linux_without_cuda.yml @@ -9,6 +9,7 @@ on: - .github/workflows/generate_binary_build_matrix.yml - .github/workflows/test_build_wheels_linux_without_cuda.yml - tools/scripts/generate_binary_build_matrix.py + - tools/pkg-helpers/pytorch_pkg_helpers/version.py workflow_dispatch: permissions: diff --git a/tools/pkg-helpers/pytorch_pkg_helpers/version.py b/tools/pkg-helpers/pytorch_pkg_helpers/version.py index 3196a6993e..edcd92a991 100644 --- a/tools/pkg-helpers/pytorch_pkg_helpers/version.py +++ b/tools/pkg-helpers/pytorch_pkg_helpers/version.py @@ -104,11 +104,7 @@ def get_version_variables( ) -> List[str]: version = PytorchVersion( gpu_arch_version=gpu_arch_version, - no_build_suffix=( - platform == "darwin" - or platform == "linux-aarch64" - or package_type == "conda" - ), + no_build_suffix=(platform == "darwin"), base_build_version=base_build_version, ) output_version = version.get_nightly_version()