[incubator-mxnet] branch v1.x updated: Fix windows dll loading for compute capabilties >7.5 (#19931)

[lausen]

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

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


The following commit(s) were added to refs/heads/v1.x by this push:
     new feed41c  Fix windows dll loading for compute capabilties >7.5 (#19931)
feed41c is described below

commit feed41cc7ea811f5ee5fab84fe25d736ba344a3f
Author: Rohit Kumar Srivastava <srivastava....@osu.edu>
AuthorDate: Mon Feb 22 15:12:00 2021 -0800

    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 <vladoovtcha...@gmail.com>
    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;
 }