fix cannot create gpu device issue

openhydra · Jul 10, 2024 · 29782d3 · 29782d3
1 parent 885b3c1
commit 29782d3
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pkg/open-hydra/device-handler.go b/pkg/open-hydra/device-handler.go
@@ -244,8 +244,11 @@ func (builder *OpenHydraRouteBuilder) DeviceCreateRouteHandler(request *restful.
  if gpuSet.Gpu > 0 {
  // go with gpu image
  if reqDevice.Spec.GpuDriver == "" {
- writeHttpResponseAndLogError(response, http.StatusBadRequest, "gpu driver is empty")
- return
+ if builder.Config.DefaultGpuDriver == "" {
+ writeHttpResponseAndLogError(response, http.StatusBadRequest, "both gpu driver and DefaultGpuDriver are empty")
+ return
+ }
+ reqDevice.Spec.GpuDriver = builder.Config.DefaultGpuDriver
  }
 
  // ensure gpu is allowed