Skip to content

Commit

Permalink
Update artifacts.auto.tfvars for CUDA 12.1 (#5683)
Browse files Browse the repository at this point in the history
Update artifacts.auto.tfvars for CUDA 12.1 for release 2.1
  • Loading branch information
ManfeiBai authored Oct 9, 2023
1 parent b84f798 commit 2d6c50a
Showing 1 changed file with 23 additions and 8 deletions.
31 changes: 23 additions & 8 deletions infra/tpu-pytorch-releases/artifacts.auto.tfvars
Original file line number Diff line number Diff line change
Expand Up @@ -35,14 +35,14 @@ xrt_versioned_builds = [
{
accelerator = "tpu"
python_version = "3.10"
pytorch_git_rev = "v2.1.0-rc6"
pytorch_git_rev = "v2.1.0"
package_version = "2.1.0+xrt"
},
{
accelerator = "cuda"
python_version = "3.10"
cuda_version = "12.0"
pytorch_git_rev = "v2.1.0-rc6"
pytorch_git_rev = "v2.1.0"
package_version = "2.1.0+xrt"
},
]
Expand All @@ -51,22 +51,22 @@ xrt_versioned_builds = [
versioned_builds = [
{
git_tag = "v2.1.0"
pytorch_git_rev = "v2.1.0-rc6"
pytorch_git_rev = "v2.1.0"
package_version = "2.1.0"
accelerator = "tpu"
bundle_libtpu = "0"
},
{
git_tag = "v2.1.0"
pytorch_git_rev = "v2.1.0-rc6"
pytorch_git_rev = "v2.1.0"
package_version = "2.1.0"
accelerator = "tpu"
python_version = "3.10"
bundle_libtpu = "0"
},
{
git_tag = "v2.1.0"
pytorch_git_rev = "v2.1.0-rc6"
pytorch_git_rev = "v2.1.0"
package_version = "2.1.0+libtpu"
accelerator = "tpu"
python_version = "3.10"
Expand All @@ -84,26 +84,41 @@ versioned_builds = [
},
{
git_tag = "v2.1.0"
pytorch_git_rev = "v2.1.0-rc6"
pytorch_git_rev = "v2.1.0"
package_version = "2.1.0",
accelerator = "cuda"
cuda_version = "12.0"
},
{
git_tag = "v2.1.0"
pytorch_git_rev = "v2.1.0-rc6"
pytorch_git_rev = "v2.1.0"
package_version = "2.1.0"
accelerator = "cuda"
cuda_version = "11.8"
},
{
git_tag = "v2.1.0"
pytorch_git_rev = "v2.1.0-rc6"
pytorch_git_rev = "v2.1.0"
package_version = "2.1.0"
accelerator = "cuda"
cuda_version = "12.1"
},
{
git_tag = "v2.1.0"
pytorch_git_rev = "v2.1.0"
package_version = "2.1.0"
accelerator = "cuda"
cuda_version = "11.8"
python_version = "3.10"
},
{
git_tag = "v2.1.0"
pytorch_git_rev = "v2.1.0"
package_version = "2.1.0"
accelerator = "cuda"
cuda_version = "12.1"
python_version = "3.10"
},
{
git_tag = "v2.0.0"
package_version = "2.0"
Expand Down

0 comments on commit 2d6c50a

Please sign in to comment.