From 29782d32f6d9ad31b25ccc1b915a722a250b6116 Mon Sep 17 00:00:00 2001 From: yangchenjun Date: Wed, 10 Jul 2024 15:44:41 +0800 Subject: [PATCH] fix cannot create gpu device issue --- pkg/open-hydra/device-handler.go | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/pkg/open-hydra/device-handler.go b/pkg/open-hydra/device-handler.go index 7eb2ecb..a8a2999 100644 --- 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