Skip getCurrentCUDABlasLtHandle

bytedeco · May 4, 2024 · db813d6 · db813d6
1 parent 04067c4
commit db813d6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/pytorch/src/gen/java/org/bytedeco/pytorch/global/torch_cuda.java b/pytorch/src/gen/java/org/bytedeco/pytorch/global/torch_cuda.java
@@ -188,7 +188,7 @@ public class torch_cuda extends org.bytedeco.pytorch.presets.torch_cuda {
 @Namespace("at::cuda") public static native @Cast("cusparseHandle_t") Pointer getCurrentCUDASparseHandle();
 @Namespace("at::cuda") public static native @Cast("cublasHandle_t") Pointer getCurrentCUDABlasHandle();
 // #if (!defined(USE_ROCM) && !defined(_MSC_VER)) || (defined(USE_ROCM) && ROCM_VERSION >= 50700)
-@Namespace("at::cuda") public static native @Cast("cublasLtHandle_t") Pointer getCurrentCUDABlasLtHandle();
+
 // #endif
 
 @Namespace("at::cuda") public static native void clearCublasWorkspaces();

diff --git a/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch_cuda.java b/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch_cuda.java
@@ -208,5 +208,10 @@ public void map(InfoMap infoMap) {
         ;
 
         infoMap.put(new Info("USE_CUDNN_RNN_V8_API").define()); // Using CuDNN 8.9.7 or more recent
+
+        //// Different C++ API between platforms
+        infoMap
+            .put(new Info("at::cuda::getCurrentCUDABlasLtHandle").skip()) // No cublas lt with Microsoft compiler
+        ;
     }
 }