[incubator-mxnet] branch master updated: Fix windows dll loading for compute capabilties >7.5 (#19410)

[zh...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

zhasheng pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/master by this push:
     new 4b3be14  Fix windows dll loading for compute capabilties >7.5 (#19410)
4b3be14 is described below

commit 4b3be14a43928473fc12d3b6f47accc59f9834c0
Author: vlado <vl...@gmail.com>
AuthorDate: Sun Oct 25 20:08:36 2020 -0600

    Fix windows dll loading for compute capabilties >7.5 (#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 <vl...@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 6a89a4e..6c27b5e 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;
 }