Merge branch 'main' of github.com:meetrajvala/go-tpm-tools into gpu-s…

…upport
google · Oct 10, 2024 · 8a29735 · 8a29735
2 parents 0ae2e12 + 6405dbb
commit 8a29735
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/...t_gpu_driver_Installation_cloudbuild.yaml → ...t_gpu_driver_installation_cloudbuild.yaml b/...t_gpu_driver_Installation_cloudbuild.yaml → ...t_gpu_driver_installation_cloudbuild.yaml
diff --git a/launcher/internal/gpu/driverinstaller.go b/launcher/internal/gpu/driverinstaller.go
@@ -23,8 +23,7 @@ const (
 	installerSnapshotID  = "tee-gpu-driver-installer-snapshot"
 )
 
-// SupportedGpuTypes is the list of supported gpu types with open sourced nvidia kernel modules.
-var SupportedGpuTypes = []deviceinfo.GPUType{
+var supportedGpuTypes = []deviceinfo.GPUType{
 	deviceinfo.L4,
 	deviceinfo.T4,
 	deviceinfo.A100_40GB,
@@ -59,7 +58,7 @@ func (di *DriverInstaller) InstallGPUDrivers(ctx context.Context) error {
 	}
 
 	if !gpuType.OpenSupported() {
-		return fmt.Errorf("unsupported gpu type %s, please retry with one of the supported gpu types: %v", gpuType.String(), gpu.SupportedGpuTypes)
+		return fmt.Errorf("unsupported gpu type %s, please retry with one of the supported gpu types: %v", gpuType.String(), supportedGpuTypes)
 	}
 
 	ctx = namespaces.WithNamespace(ctx, namespaces.Default)
@@ -132,7 +131,7 @@ func (di *DriverInstaller) InstallGPUDrivers(ctx context.Context) error {
 	code, _, _ := status.Result()
 	di.logger.Printf("Gpu driver installation task exited with status: %d\n", code)
 
-	err = remountAsExecutable(gpu.InstallationHostDir)
+	err = remountAsExecutable(InstallationHostDir)
 	if err != nil {
 		return fmt.Errorf("failed to remount the installed drivers: %v", err)
 	}