Make nvidia version data optional for ROCm builds

phambinhfin · phambinhfin · commit 33e668f91f9d · 2025-11-07T16:48:32.000-06:00
(cherry picked from commit 730790a)
diff --git a/jaxlib/tools/build_gpu_kernels_wheel.py b/jaxlib/tools/build_gpu_kernels_wheel.py
@@ -67,7 +67,7 @@
 parser.add_argument(
     "--nvidia_wheel_versions_data",
     default=None,
-    required=True,
+    required=False,
     help="NVIDIA wheel versions data",
 )
 args = parser.parse_args()
diff --git a/jaxlib/tools/build_gpu_plugin_wheel.py b/jaxlib/tools/build_gpu_plugin_wheel.py
@@ -73,7 +73,7 @@
 parser.add_argument(
     "--nvidia_wheel_versions_data",
     default=None,
-    required=True,
+    required=False,
     help="NVIDIA wheel versions data",
 )
 args = parser.parse_args()

Original file line number	Diff line number	Diff line change
`@@ -67,7 +67,7 @@`
`67`	`67`	`parser.add_argument(`
`68`	`68`	`"--nvidia_wheel_versions_data",`
`69`	`69`	`default=None,`
`70`		`- required=True,`
	`70`	`+ required=False,`
`71`	`71`	`help="NVIDIA wheel versions data",`
`72`	`72`	`)`
`73`	`73`	`args = parser.parse_args()`
Original file line number	Diff line number	Diff line change
`@@ -73,7 +73,7 @@`
`73`	`73`	`parser.add_argument(`
`74`	`74`	`"--nvidia_wheel_versions_data",`
`75`	`75`	`default=None,`
`76`		`- required=True,`
	`76`	`+ required=False,`
`77`	`77`	`help="NVIDIA wheel versions data",`
`78`	`78`	`)`
`79`	`79`	`args = parser.parse_args()`