Add cupti from cuda presets

HGuillemet · Aug 21, 2024 · e897fef · e897fef
1 parent ac9e231
commit e897fef
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch.java b/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch.java
@@ -205,6 +205,7 @@ public void init(ClassProperties properties) {
                     : lib.equals("nvrtc") ? "64_120_0"
                     : lib.equals("nvrtc-builtins") ? "64_126"
                     : lib.equals("nvJitLink") ? "_120_0"
+                    : lib.equals("cupti") ? "64_2024.3.0"
                     : "64_12";
             } else {
                 continue; // no CUDA

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
@@ -22,6 +22,7 @@
 package org.bytedeco.pytorch.presets;
 
 import org.bytedeco.cuda.presets.cudnn;
+import org.bytedeco.cuda.presets.cupti;
 import org.bytedeco.cuda.presets.cusolver;
 import org.bytedeco.cuda.presets.cusparse;
 import org.bytedeco.javacpp.ClassProperties;
@@ -36,7 +37,7 @@
  * @author Hervé Guillemet
  */
 @Properties(
-    inherit = { torch.class, cudnn.class, cusparse.class, cusolver.class },
+    inherit = { torch.class, cudnn.class, cusparse.class, cusolver.class, cupti.class },
     value = {
         @Platform(
             extension = "-gpu",