From f29b167e1af8a7c0e8a15044584826432aff76d8 Mon Sep 17 00:00:00 2001
From: Daniel Hiltgen <dhiltgen@users.noreply.github.com>
Date: Tue, 3 Sep 2024 17:15:31 -0700
Subject: [PATCH] Use cuda v11 for driver 525 and older (#6620)

It looks like driver 525 (aka, cuda driver 12.0) has problems with the cuda v12 library
we compile against, so run v11 on those older drivers if detected.
---
 gpu/cuda_common.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gpu/cuda_common.go b/gpu/cuda_common.go
index 827cc9b4..aceec70a 100644
--- a/gpu/cuda_common.go
+++ b/gpu/cuda_common.go
@@ -57,7 +57,7 @@ func cudaVariant(gpuInfo CudaGPUInfo) string {
 		}
 	}
 
-	if gpuInfo.computeMajor < 6 || gpuInfo.DriverMajor < 12 {
+	if gpuInfo.computeMajor < 6 || gpuInfo.DriverMajor < 12 || (gpuInfo.DriverMajor == 12 && gpuInfo.DriverMinor == 0) {
 		return "v11"
 	}
 	return "v12"