Ensure zlibwapi.dll gets preloaded in presets that need it for cuDN…

…N (issue bytedeco/javacv#1767)
bytedeco · Mar 8, 2022 · 24846f5 · 24846f5
1 parent 3db41e8
commit 24846f5
Show file tree

Hide file tree

Showing 8 changed files with 8 additions and 8 deletions.
diff --git a/caffe/src/main/java/org/bytedeco/caffe/presets/caffe.java b/caffe/src/main/java/org/bytedeco/caffe/presets/caffe.java
@@ -80,7 +80,7 @@ public class caffe implements LoadEnabled, InfoMapper {
             return;
         }
         int i = 0;
-        String[] libs = {"cudart", "cublasLt", "cublas", "curand", "cudnn",
+        String[] libs = {"cudart", "cublasLt", "cublas", "curand", "cudnn", "zlibwapi",
                          "cudnn_ops_infer", "cudnn_ops_train", "cudnn_adv_infer",
                          "cudnn_adv_train", "cudnn_cnn_infer", "cudnn_cnn_train"};
         for (String lib : libs) {

diff --git a/mxnet/src/main/java/org/bytedeco/mxnet/presets/mxnet.java b/mxnet/src/main/java/org/bytedeco/mxnet/presets/mxnet.java
@@ -77,7 +77,7 @@ public class mxnet implements LoadEnabled, InfoMapper {
             return;
         }
         int i = 0;
-        String[] libs = {"cudart", "cublasLt", "cublas", "cufft", "curand", "cusolver", "cudnn", "nccl", "nvrtc",
+        String[] libs = {"cudart", "cublasLt", "cublas", "cufft", "curand", "cusolver", "cudnn", "nccl", "nvrtc", "zlibwapi",
                          "cudnn_ops_infer", "cudnn_ops_train", "cudnn_adv_infer", "cudnn_adv_train", "cudnn_cnn_infer", "cudnn_cnn_train"};
         for (String lib : libs) {
             if (platform.startsWith("linux")) {

diff --git a/onnxruntime/src/main/java/org/bytedeco/onnxruntime/presets/onnxruntime.java b/onnxruntime/src/main/java/org/bytedeco/onnxruntime/presets/onnxruntime.java
@@ -90,7 +90,7 @@ public class onnxruntime implements LoadEnabled, InfoMapper {
             return;
         }
         int i = 0;
-        String[] libs = {"cudart", "cublasLt", "cublas", "cufft", "curand", "cudnn",
+        String[] libs = {"cudart", "cublasLt", "cublas", "cufft", "curand", "cudnn", "zlibwapi",
                          "cudnn_ops_infer", "cudnn_ops_train", "cudnn_adv_infer",
                          "cudnn_adv_train", "cudnn_cnn_infer", "cudnn_cnn_train"};
         for (String lib : libs) {

diff --git a/opencv/src/main/java/org/bytedeco/opencv/presets/opencv_core.java b/opencv/src/main/java/org/bytedeco/opencv/presets/opencv_core.java
@@ -90,7 +90,7 @@ public class opencv_core implements LoadEnabled, InfoMapper {
         }
         int i = 0;
         String[] libs = {"cudart", "cublasLt", "cublas", "cufft", "cudnn", "nppc", "nppial", "nppicc", "nppicom",
-                         "nppidei", "nppif", "nppig", "nppim", "nppist", "nppisu", "nppitc", "npps",
+                         "nppidei", "nppif", "nppig", "nppim", "nppist", "nppisu", "nppitc", "npps", "zlibwapi",
                          "cudnn_ops_infer", "cudnn_ops_train", "cudnn_adv_infer", "cudnn_adv_train", "cudnn_cnn_infer", "cudnn_cnn_train"};
         for (String lib : libs) {
             if (platform.startsWith("linux")) {

diff --git a/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch.java b/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch.java
@@ -458,7 +458,7 @@ public class torch implements LoadEnabled, InfoMapper {
         }
         int i = 0;
         String[] libs = {"cudart", "cublasLt", "cublas", "cufft", "curand", "cusolver", "cusparse", "cudnn", "nccl", "nvrtc", "myelin", "nvinfer",
-                         "cudnn_ops_infer", "cudnn_ops_train", "cudnn_adv_infer", "cudnn_adv_train", "cudnn_cnn_infer", "cudnn_cnn_train"};
+                         "zlibwapi", "cudnn_ops_infer", "cudnn_ops_train", "cudnn_adv_infer", "cudnn_adv_train", "cudnn_cnn_infer", "cudnn_cnn_train"};
         for (String lib : libs) {
             if (platform.startsWith("linux")) {
                 lib += lib.startsWith("cudnn") ? "@.8"

diff --git a/tensorflow/src/main/java/org/bytedeco/tensorflow/presets/tensorflow.java b/tensorflow/src/main/java/org/bytedeco/tensorflow/presets/tensorflow.java
@@ -511,7 +511,7 @@ public void init(Logger logger, java.util.Properties properties, String encoding
             return;
         }
         String[] libs = {"cudart", "cublasLt", "cublas", "cufft", "curand", "cusolver", "cusparse", "cudnn", "nccl", "nvrtc", "myelin", "nvinfer",
-                         "cudnn_ops_infer", "cudnn_ops_train", "cudnn_adv_infer", "cudnn_adv_train", "cudnn_cnn_infer", "cudnn_cnn_train"};
+                         "zlibwapi", "cudnn_ops_infer", "cudnn_ops_train", "cudnn_adv_infer", "cudnn_adv_train", "cudnn_cnn_infer", "cudnn_cnn_train"};
         for (String lib : libs) {
             if (platform.startsWith("linux")) {
                 lib += lib.startsWith("cudnn") ? "@.8"

diff --git a/tensorrt/src/main/java/org/bytedeco/tensorrt/presets/nvinfer.java b/tensorrt/src/main/java/org/bytedeco/tensorrt/presets/nvinfer.java
@@ -88,7 +88,7 @@ public class nvinfer implements LoadEnabled, InfoMapper {
             return;
         }
         int i = 0;
-        String[] libs = {"cudart", "cublasLt", "cublas", "cudnn", "nvrtc",
+        String[] libs = {"cudart", "cublasLt", "cublas", "cudnn", "nvrtc", "zlibwapi",
                          "cudnn_ops_infer", "cudnn_ops_train", "cudnn_adv_infer",
                          "cudnn_adv_train", "cudnn_cnn_infer", "cudnn_cnn_train"};
         for (String lib : libs) {

diff --git a/tvm/src/main/java/org/bytedeco/tvm/presets/tvm_runtime.java b/tvm/src/main/java/org/bytedeco/tvm/presets/tvm_runtime.java
@@ -115,7 +115,7 @@ public class tvm_runtime implements LoadEnabled, InfoMapper {
             return;
         }
         int i = 0;
-        String[] libs = {"cudart", "cublasLt", "cublas", "cudnn", "nvrtc",
+        String[] libs = {"cudart", "cublasLt", "cublas", "cudnn", "nvrtc", "zlibwapi",
                          "cudnn_ops_infer", "cudnn_ops_train", "cudnn_adv_infer",
                          "cudnn_adv_train", "cudnn_cnn_infer", "cudnn_cnn_train"};
         for (String lib : libs) {