Fix windows dll loading for compute capabilties >7.5 (#19931)

Backport #19410 Previously any higher compute capabilties would load mxnet_75.dll Any new compute capabilities should now load the correct dll if present Co-authored-by: vlado <[email protected]> Co-authored-by: vlado <[email protected]>
apache · Feb 22, 2021 · feed41c · feed41c
1 parent c97e89c
commit feed41c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tools/windowsbuild/warp_dll.cpp b/tools/windowsbuild/warp_dll.cpp
@@ -78,7 +78,7 @@ int find_version()
 {
 	std::vector<int> known_sm = find_mxnet_dll();
 	int count = 0;
-	int version = 75;
+	int version = 9999;
 	if (cudaSuccess != cudaGetDeviceCount(&count))
 	{
 		return 30;
@@ -106,6 +106,11 @@ int find_version()
 		}
 	}
 
+    if (version == 9999)
+    {
+        return 30;
+    }
+
 	return version;
 }