From feed41cc7ea811f5ee5fab84fe25d736ba344a3f Mon Sep 17 00:00:00 2001
From: Rohit Kumar Srivastava <srivastava.141@osu.edu>
Date: Mon, 22 Feb 2021 15:12:00 -0800
Subject: [PATCH] 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 <vladoovtcharov@gmail.com>
Co-authored-by: vlado <vlado@indicalab.com>
---
 tools/windowsbuild/warp_dll.cpp | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/tools/windowsbuild/warp_dll.cpp b/tools/windowsbuild/warp_dll.cpp
index 6a89a4e189de..6c27b5e48598 100644
--- 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;
 }