Update libtpu (#906)

wsmoses · web-flow · commit 28093da3a053 · 2025-03-13T17:07:05.000-05:00
diff --git a/src/xla/XLA.jl b/src/xla/XLA.jl
@@ -228,7 +228,7 @@ end
 function download_tpu(dataset_dir::String)
     if !isfile(dataset_dir * "/libtpu.so")
         Downloads.download(
-            "https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/wheels/libtpu-nightly/libtpu_nightly-0.1.dev20240829-py3-none-any.whl",
+            "https://storage.googleapis.com/libtpu-nightly-releases/wheels/libtpu-nightly/libtpu_nightly-0.1.dev20250313+nightly-py3-none-manylinux_2_31_x86_64.whl",
             dataset_dir * "/tpu.zip",
         )
         run(`unzip -qq $(dataset_dir*"/tpu.zip") -d $(dataset_dir)/tmp`)

Original file line number	Diff line number	Diff line change
`@@ -228,7 +228,7 @@ end`
`228`	`228`	`function download_tpu(dataset_dir::String)`
`229`	`229`	`if !isfile(dataset_dir * "/libtpu.so")`
`230`	`230`	`Downloads.download(`
`231`		`- "https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/wheels/libtpu-nightly/libtpu_nightly-0.1.dev20240829-py3-none-any.whl",`
	`231`	`+ "https://storage.googleapis.com/libtpu-nightly-releases/wheels/libtpu-nightly/libtpu_nightly-0.1.dev20250313+nightly-py3-none-manylinux_2_31_x86_64.whl",`
`232`	`232`	`dataset_dir * "/tpu.zip",`
`233`	`233`	`)`
`234`	`234`	run(`unzip -qq $(dataset_dir*"/tpu.zip") -d $(dataset_dir)/tmp`)