update nightly version number (pytorch#6842)

Co-authored-by: Siyuan Liu <lsiyuan@google.coim>
lsy323 · Mar 27, 2024 · f6d6286 · f6d6286
1 parent 59db0cd
commit f6d6286
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/infra/ansible/config/vars.yaml b/infra/ansible/config/vars.yaml
@@ -5,7 +5,7 @@ cuda_version: "11.8"
 llvm_debian_repo: bullseye
 clang_version: 17
 # PyTorch and PyTorch/XLA wheel versions.
-package_version: 2.3.0
+package_version: 2.4.0
 # If set to true, wheels will be renamed to $WHEEL_NAME-nightly-cp38-cp38-linux_x86_64.whl.
 nightly_release: false
 # Whether to preinstall libtpu in the PyTorch/XLA wheel. Ignored for GPU build.

diff --git a/infra/tpu-pytorch-releases/artifacts.auto.tfvars b/infra/tpu-pytorch-releases/artifacts.auto.tfvars
@@ -1,4 +1,4 @@
-nightly_package_version = "2.3.0"
+nightly_package_version = "2.4.0"
 
 # Built once a day from master.
 nightly_builds = [

diff --git a/setup.py b/setup.py
@@ -90,7 +90,7 @@ def get_git_head_sha(base_dir):
 
 
 def get_build_version(xla_git_sha):
-  version = os.getenv('TORCH_XLA_VERSION', '2.3.0')
+  version = os.getenv('TORCH_XLA_VERSION', '2.4.0')
   if build_util.check_env_flag('GIT_VERSIONED_XLA_BUILD', default='TRUE'):
     try:
       version += '+git' + xla_git_sha[:7]