Maintain the requirement for CUDA builds

phambinhfin · phambinhfin · commit b8bc25465b55 · 2025-11-13T16:28:24.000-06:00
diff --git a/jaxlib/tools/build_gpu_kernels_wheel.py b/jaxlib/tools/build_gpu_kernels_wheel.py
@@ -72,6 +72,9 @@
 )
 args = parser.parse_args()
 
+if args.enable_cuda and not args.nvidia_wheel_versions_data:
+  parser.error('argument --nvidia_wheel_versions_data is required for CUDA builds')
+
 r = runfiles.Create()
 pyext = "pyd" if build_utils.is_windows() else "so"
 
diff --git a/jaxlib/tools/build_gpu_plugin_wheel.py b/jaxlib/tools/build_gpu_plugin_wheel.py
@@ -77,7 +77,8 @@
     help="NVIDIA wheel versions data",
 )
 args = parser.parse_args()
-
+if args.enable_cuda and not args.nvidia_wheel_versions_data:
+  parser.error('argument --nvidia_wheel_versions_data is required for CUDA builds')
 r = runfiles.Create()
 
 

Original file line number	Diff line number	Diff line change
`@@ -72,6 +72,9 @@`
`72`	`72`	`)`
`73`	`73`	`args = parser.parse_args()`
`74`	`74`
	`75`	`+if args.enable_cuda and not args.nvidia_wheel_versions_data:`
	`76`	`+ parser.error('argument --nvidia_wheel_versions_data is required for CUDA builds')`
	`77`	`+`
`75`	`78`	`r = runfiles.Create()`
`76`	`79`	`pyext = "pyd" if build_utils.is_windows() else "so"`
`77`	`80`
Original file line number	Diff line number	Diff line change
`@@ -77,7 +77,8 @@`
`77`	`77`	`help="NVIDIA wheel versions data",`
`78`	`78`	`)`
`79`	`79`	`args = parser.parse_args()`
`80`		`-`
	`80`	`+if args.enable_cuda and not args.nvidia_wheel_versions_data:`
	`81`	`+ parser.error('argument --nvidia_wheel_versions_data is required for CUDA builds')`
`81`	`82`	`r = runfiles.Create()`
`82`	`83`
`83`	`84`