Make torch presets inherit from cuda presets

bytedeco · Aug 1, 2024 · 1f57f8c · 1f57f8c
1 parent e372f94
commit 1f57f8c
Show file tree

Hide file tree

Showing 45 changed files with 417 additions and 226 deletions.
diff --git a/pytorch/src/gen/java/org/bytedeco/pytorch/AutogradCompilerCall.java b/pytorch/src/gen/java/org/bytedeco/pytorch/AutogradCompilerCall.java
@@ -42,9 +42,9 @@ public class AutogradCompilerCall extends Pointer {
   public native @Cast("size_t") long emplace_hook(@ByRef(true) SafePyObject fn);
 
   public native @ByRef TensorArgs tensor_args(); public native AutogradCompilerCall tensor_args(TensorArgs setter);
-  public native @StdVector SizeInput all_size_inputs(); public native AutogradCompilerCall all_size_inputs(SizeInput setter);
-  public native @ByRef @Cast("std::vector<int64_t>*") LongVector dyn_size_inputs(); public native AutogradCompilerCall dyn_size_inputs(LongVector setter);
+  public native @StdVector @NoOffset SizeInput all_size_inputs(); public native AutogradCompilerCall all_size_inputs(SizeInput setter);
+  public native @ByRef @Cast("std::vector<int64_t>*") @NoOffset LongVector dyn_size_inputs(); public native AutogradCompilerCall dyn_size_inputs(LongVector setter);
 
-  public native @ByRef NodeCalls node_calls(); public native AutogradCompilerCall node_calls(NodeCalls setter);
-  public native SizeInput.DynType default_dyn_type(); public native AutogradCompilerCall default_dyn_type(SizeInput.DynType setter);
+  public native @ByRef @NoOffset NodeCalls node_calls(); public native AutogradCompilerCall node_calls(NodeCalls setter);
+  public native @NoOffset SizeInput.DynType default_dyn_type(); public native AutogradCompilerCall default_dyn_type(SizeInput.DynType setter);
 }
diff --git a/pytorch/src/gen/java/org/bytedeco/pytorch/C10dLoggingData.java b/pytorch/src/gen/java/org/bytedeco/pytorch/C10dLoggingData.java
@@ -41,7 +41,7 @@ public class C10dLoggingData extends Pointer {
     }
 
   // logging fields that are string types.
-  public native @ByRef StringStringMap strings(); public native C10dLoggingData strings(StringStringMap setter);
+  public native @ByRef @NoOffset StringStringMap strings(); public native C10dLoggingData strings(StringStringMap setter);
   // logging fields that are int64_t types.
-  public native @ByRef StringLongMap integers(); public native C10dLoggingData integers(StringLongMap setter);
+  public native @ByRef @NoOffset StringLongMap integers(); public native C10dLoggingData integers(StringLongMap setter);
 }
diff --git a/pytorch/src/gen/java/org/bytedeco/pytorch/TensorArgs.java b/pytorch/src/gen/java/org/bytedeco/pytorch/TensorArgs.java
@@ -47,5 +47,5 @@ public class TensorArgs extends Pointer {
   public native @ByRef DynamoTensorArg add(@Const @ByRef Tensor tensor);
 
   // the concrete tensors that will get passed into the graph as inputs
-  public native @ByRef TensorVector inputs(); public native TensorArgs inputs(TensorVector setter);
+  public native @ByRef @NoOffset TensorVector inputs(); public native TensorArgs inputs(TensorVector setter);
 }
diff --git a/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/AOTIModelContainerRunnerCuda.java b/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/AOTIModelContainerRunnerCuda.java
@@ -4,11 +4,6 @@
 
 import org.bytedeco.pytorch.*;
 import org.bytedeco.pytorch.helper.*;
-import org.bytedeco.cuda.cudart.*;
-import org.bytedeco.cuda.cusparse.*;
-import org.bytedeco.cuda.cublas.*;
-import org.bytedeco.cuda.cusolver.*;
-import org.bytedeco.cuda.cudnn.*;
 import org.bytedeco.pytorch.chrono.*;
 import org.bytedeco.pytorch.global.torch.DeviceType;
 import org.bytedeco.pytorch.global.torch.ScalarType;
@@ -23,6 +18,16 @@
 import static org.bytedeco.openblas.global.openblas.*;
 import org.bytedeco.pytorch.*;
 import static org.bytedeco.pytorch.global.torch.*;
+import org.bytedeco.cuda.cudart.*;
+import static org.bytedeco.cuda.global.cudart.*;
+import org.bytedeco.cuda.cublas.*;
+import static org.bytedeco.cuda.global.cublas.*;
+import org.bytedeco.cuda.cudnn.*;
+import static org.bytedeco.cuda.global.cudnn.*;
+import org.bytedeco.cuda.cusparse.*;
+import static org.bytedeco.cuda.global.cusparse.*;
+import org.bytedeco.cuda.cusolver.*;
+import static org.bytedeco.cuda.global.cusolver.*;
 
 import static org.bytedeco.pytorch.global.torch_cuda.*;
 

diff --git a/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/ActivationDescriptor.java b/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/ActivationDescriptor.java
@@ -4,11 +4,6 @@
 
 import org.bytedeco.pytorch.*;
 import org.bytedeco.pytorch.helper.*;
-import org.bytedeco.cuda.cudart.*;
-import org.bytedeco.cuda.cusparse.*;
-import org.bytedeco.cuda.cublas.*;
-import org.bytedeco.cuda.cusolver.*;
-import org.bytedeco.cuda.cudnn.*;
 import org.bytedeco.pytorch.chrono.*;
 import org.bytedeco.pytorch.global.torch.DeviceType;
 import org.bytedeco.pytorch.global.torch.ScalarType;
@@ -23,6 +18,16 @@
 import static org.bytedeco.openblas.global.openblas.*;
 import org.bytedeco.pytorch.*;
 import static org.bytedeco.pytorch.global.torch.*;
+import org.bytedeco.cuda.cudart.*;
+import static org.bytedeco.cuda.global.cudart.*;
+import org.bytedeco.cuda.cublas.*;
+import static org.bytedeco.cuda.global.cublas.*;
+import org.bytedeco.cuda.cudnn.*;
+import static org.bytedeco.cuda.global.cudnn.*;
+import org.bytedeco.cuda.cusparse.*;
+import static org.bytedeco.cuda.global.cusparse.*;
+import org.bytedeco.cuda.cusolver.*;
+import static org.bytedeco.cuda.global.cusolver.*;
 
 import static org.bytedeco.pytorch.global.torch_cuda.*;
 

diff --git a/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/AllocatorConfigInfo.java b/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/AllocatorConfigInfo.java
@@ -4,11 +4,6 @@
 
 import org.bytedeco.pytorch.*;
 import org.bytedeco.pytorch.helper.*;
-import org.bytedeco.cuda.cudart.*;
-import org.bytedeco.cuda.cusparse.*;
-import org.bytedeco.cuda.cublas.*;
-import org.bytedeco.cuda.cusolver.*;
-import org.bytedeco.cuda.cudnn.*;
 import org.bytedeco.pytorch.chrono.*;
 import org.bytedeco.pytorch.global.torch.DeviceType;
 import org.bytedeco.pytorch.global.torch.ScalarType;
@@ -23,6 +18,16 @@
 import static org.bytedeco.openblas.global.openblas.*;
 import org.bytedeco.pytorch.*;
 import static org.bytedeco.pytorch.global.torch.*;
+import org.bytedeco.cuda.cudart.*;
+import static org.bytedeco.cuda.global.cudart.*;
+import org.bytedeco.cuda.cublas.*;
+import static org.bytedeco.cuda.global.cublas.*;
+import org.bytedeco.cuda.cudnn.*;
+import static org.bytedeco.cuda.global.cudnn.*;
+import org.bytedeco.cuda.cusparse.*;
+import static org.bytedeco.cuda.global.cusparse.*;
+import org.bytedeco.cuda.cusolver.*;
+import static org.bytedeco.cuda.global.cusolver.*;
 
 import static org.bytedeco.pytorch.global.torch_cuda.*;
 

diff --git a/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/AllocatorState.java b/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/AllocatorState.java
@@ -4,11 +4,6 @@
 
 import org.bytedeco.pytorch.*;
 import org.bytedeco.pytorch.helper.*;
-import org.bytedeco.cuda.cudart.*;
-import org.bytedeco.cuda.cusparse.*;
-import org.bytedeco.cuda.cublas.*;
-import org.bytedeco.cuda.cusolver.*;
-import org.bytedeco.cuda.cudnn.*;
 import org.bytedeco.pytorch.chrono.*;
 import org.bytedeco.pytorch.global.torch.DeviceType;
 import org.bytedeco.pytorch.global.torch.ScalarType;
@@ -23,6 +18,16 @@
 import static org.bytedeco.openblas.global.openblas.*;
 import org.bytedeco.pytorch.*;
 import static org.bytedeco.pytorch.global.torch.*;
+import org.bytedeco.cuda.cudart.*;
+import static org.bytedeco.cuda.global.cudart.*;
+import org.bytedeco.cuda.cublas.*;
+import static org.bytedeco.cuda.global.cublas.*;
+import org.bytedeco.cuda.cudnn.*;
+import static org.bytedeco.cuda.global.cudnn.*;
+import org.bytedeco.cuda.cusparse.*;
+import static org.bytedeco.cuda.global.cusparse.*;
+import org.bytedeco.cuda.cusolver.*;
+import static org.bytedeco.cuda.global.cusolver.*;
 
 import static org.bytedeco.pytorch.global.torch_cuda.*;
 

diff --git a/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/BlockInfo.java b/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/BlockInfo.java
@@ -4,11 +4,6 @@
 
 import org.bytedeco.pytorch.*;
 import org.bytedeco.pytorch.helper.*;
-import org.bytedeco.cuda.cudart.*;
-import org.bytedeco.cuda.cusparse.*;
-import org.bytedeco.cuda.cublas.*;
-import org.bytedeco.cuda.cusolver.*;
-import org.bytedeco.cuda.cudnn.*;
 import org.bytedeco.pytorch.chrono.*;
 import org.bytedeco.pytorch.global.torch.DeviceType;
 import org.bytedeco.pytorch.global.torch.ScalarType;
@@ -23,6 +18,16 @@
 import static org.bytedeco.openblas.global.openblas.*;
 import org.bytedeco.pytorch.*;
 import static org.bytedeco.pytorch.global.torch.*;
+import org.bytedeco.cuda.cudart.*;
+import static org.bytedeco.cuda.global.cudart.*;
+import org.bytedeco.cuda.cublas.*;
+import static org.bytedeco.cuda.global.cublas.*;
+import org.bytedeco.cuda.cudnn.*;
+import static org.bytedeco.cuda.global.cudnn.*;
+import org.bytedeco.cuda.cusparse.*;
+import static org.bytedeco.cuda.global.cusparse.*;
+import org.bytedeco.cuda.cusolver.*;
+import static org.bytedeco.cuda.global.cusolver.*;
 
 import static org.bytedeco.pytorch.global.torch_cuda.*;
 

diff --git a/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/CTCLossDescriptor.java b/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/CTCLossDescriptor.java
@@ -4,11 +4,6 @@
 
 import org.bytedeco.pytorch.*;
 import org.bytedeco.pytorch.helper.*;
-import org.bytedeco.cuda.cudart.*;
-import org.bytedeco.cuda.cusparse.*;
-import org.bytedeco.cuda.cublas.*;
-import org.bytedeco.cuda.cusolver.*;
-import org.bytedeco.cuda.cudnn.*;
 import org.bytedeco.pytorch.chrono.*;
 import org.bytedeco.pytorch.global.torch.DeviceType;
 import org.bytedeco.pytorch.global.torch.ScalarType;
@@ -23,6 +18,16 @@
 import static org.bytedeco.openblas.global.openblas.*;
 import org.bytedeco.pytorch.*;
 import static org.bytedeco.pytorch.global.torch.*;
+import org.bytedeco.cuda.cudart.*;
+import static org.bytedeco.cuda.global.cudart.*;
+import org.bytedeco.cuda.cublas.*;
+import static org.bytedeco.cuda.global.cublas.*;
+import org.bytedeco.cuda.cudnn.*;
+import static org.bytedeco.cuda.global.cudnn.*;
+import org.bytedeco.cuda.cusparse.*;
+import static org.bytedeco.cuda.global.cusparse.*;
+import org.bytedeco.cuda.cusolver.*;
+import static org.bytedeco.cuda.global.cusolver.*;
 
 import static org.bytedeco.pytorch.global.torch_cuda.*;
 

diff --git a/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/CUDAAllocator.java b/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/CUDAAllocator.java
@@ -4,11 +4,6 @@
 
 import org.bytedeco.pytorch.*;
 import org.bytedeco.pytorch.helper.*;
-import org.bytedeco.cuda.cudart.*;
-import org.bytedeco.cuda.cusparse.*;
-import org.bytedeco.cuda.cublas.*;
-import org.bytedeco.cuda.cusolver.*;
-import org.bytedeco.cuda.cudnn.*;
 import org.bytedeco.pytorch.chrono.*;
 import org.bytedeco.pytorch.global.torch.DeviceType;
 import org.bytedeco.pytorch.global.torch.ScalarType;
@@ -23,6 +18,16 @@
 import static org.bytedeco.openblas.global.openblas.*;
 import org.bytedeco.pytorch.*;
 import static org.bytedeco.pytorch.global.torch.*;
+import org.bytedeco.cuda.cudart.*;
+import static org.bytedeco.cuda.global.cudart.*;
+import org.bytedeco.cuda.cublas.*;
+import static org.bytedeco.cuda.global.cublas.*;
+import org.bytedeco.cuda.cudnn.*;
+import static org.bytedeco.cuda.global.cudnn.*;
+import org.bytedeco.cuda.cusparse.*;
+import static org.bytedeco.cuda.global.cusparse.*;
+import org.bytedeco.cuda.cusolver.*;
+import static org.bytedeco.cuda.global.cusolver.*;
 
 import static org.bytedeco.pytorch.global.torch_cuda.*;
 

diff --git a/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/CUDAEvent.java b/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/CUDAEvent.java
@@ -4,11 +4,6 @@
 
 import org.bytedeco.pytorch.*;
 import org.bytedeco.pytorch.helper.*;
-import org.bytedeco.cuda.cudart.*;
-import org.bytedeco.cuda.cusparse.*;
-import org.bytedeco.cuda.cublas.*;
-import org.bytedeco.cuda.cusolver.*;
-import org.bytedeco.cuda.cudnn.*;
 import org.bytedeco.pytorch.chrono.*;
 import org.bytedeco.pytorch.global.torch.DeviceType;
 import org.bytedeco.pytorch.global.torch.ScalarType;
@@ -23,6 +18,16 @@
 import static org.bytedeco.openblas.global.openblas.*;
 import org.bytedeco.pytorch.*;
 import static org.bytedeco.pytorch.global.torch.*;
+import org.bytedeco.cuda.cudart.*;
+import static org.bytedeco.cuda.global.cudart.*;
+import org.bytedeco.cuda.cublas.*;
+import static org.bytedeco.cuda.global.cublas.*;
+import org.bytedeco.cuda.cudnn.*;
+import static org.bytedeco.cuda.global.cudnn.*;
+import org.bytedeco.cuda.cusparse.*;
+import static org.bytedeco.cuda.global.cusparse.*;
+import org.bytedeco.cuda.cusolver.*;
+import static org.bytedeco.cuda.global.cusolver.*;
 
 import static org.bytedeco.pytorch.global.torch_cuda.*;
 

diff --git a/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/CUDAGuard.java b/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/CUDAGuard.java
@@ -4,11 +4,6 @@
 
 import org.bytedeco.pytorch.*;
 import org.bytedeco.pytorch.helper.*;
-import org.bytedeco.cuda.cudart.*;
-import org.bytedeco.cuda.cusparse.*;
-import org.bytedeco.cuda.cublas.*;
-import org.bytedeco.cuda.cusolver.*;
-import org.bytedeco.cuda.cudnn.*;
 import org.bytedeco.pytorch.chrono.*;
 import org.bytedeco.pytorch.global.torch.DeviceType;
 import org.bytedeco.pytorch.global.torch.ScalarType;
@@ -23,6 +18,16 @@
 import static org.bytedeco.openblas.global.openblas.*;
 import org.bytedeco.pytorch.*;
 import static org.bytedeco.pytorch.global.torch.*;
+import org.bytedeco.cuda.cudart.*;
+import static org.bytedeco.cuda.global.cudart.*;
+import org.bytedeco.cuda.cublas.*;
+import static org.bytedeco.cuda.global.cublas.*;
+import org.bytedeco.cuda.cudnn.*;
+import static org.bytedeco.cuda.global.cudnn.*;
+import org.bytedeco.cuda.cusparse.*;
+import static org.bytedeco.cuda.global.cusparse.*;
+import org.bytedeco.cuda.cusolver.*;
+import static org.bytedeco.cuda.global.cusolver.*;
 
 import static org.bytedeco.pytorch.global.torch_cuda.*;
 

diff --git a/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/CUDAKernelLaunchInfo.java b/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/CUDAKernelLaunchInfo.java
@@ -4,11 +4,6 @@
 
 import org.bytedeco.pytorch.*;
 import org.bytedeco.pytorch.helper.*;
-import org.bytedeco.cuda.cudart.*;
-import org.bytedeco.cuda.cusparse.*;
-import org.bytedeco.cuda.cublas.*;
-import org.bytedeco.cuda.cusolver.*;
-import org.bytedeco.cuda.cudnn.*;
 import org.bytedeco.pytorch.chrono.*;
 import org.bytedeco.pytorch.global.torch.DeviceType;
 import org.bytedeco.pytorch.global.torch.ScalarType;
@@ -23,6 +18,16 @@
 import static org.bytedeco.openblas.global.openblas.*;
 import org.bytedeco.pytorch.*;
 import static org.bytedeco.pytorch.global.torch.*;
+import org.bytedeco.cuda.cudart.*;
+import static org.bytedeco.cuda.global.cudart.*;
+import org.bytedeco.cuda.cublas.*;
+import static org.bytedeco.cuda.global.cublas.*;
+import org.bytedeco.cuda.cudnn.*;
+import static org.bytedeco.cuda.global.cudnn.*;
+import org.bytedeco.cuda.cusparse.*;
+import static org.bytedeco.cuda.global.cusparse.*;
+import org.bytedeco.cuda.cusolver.*;
+import static org.bytedeco.cuda.global.cusolver.*;
 
 import static org.bytedeco.pytorch.global.torch_cuda.*;
 

diff --git a/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/CUDAKernelLaunchInfoVector.java b/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/CUDAKernelLaunchInfoVector.java
@@ -4,11 +4,6 @@
 
 import org.bytedeco.pytorch.*;
 import org.bytedeco.pytorch.helper.*;
-import org.bytedeco.cuda.cudart.*;
-import org.bytedeco.cuda.cusparse.*;
-import org.bytedeco.cuda.cublas.*;
-import org.bytedeco.cuda.cusolver.*;
-import org.bytedeco.cuda.cudnn.*;
 import org.bytedeco.pytorch.chrono.*;
 import org.bytedeco.pytorch.global.torch.DeviceType;
 import org.bytedeco.pytorch.global.torch.ScalarType;
@@ -23,6 +18,16 @@
 import static org.bytedeco.openblas.global.openblas.*;
 import org.bytedeco.pytorch.*;
 import static org.bytedeco.pytorch.global.torch.*;
+import org.bytedeco.cuda.cudart.*;
+import static org.bytedeco.cuda.global.cudart.*;
+import org.bytedeco.cuda.cublas.*;
+import static org.bytedeco.cuda.global.cublas.*;
+import org.bytedeco.cuda.cudnn.*;
+import static org.bytedeco.cuda.global.cudnn.*;
+import org.bytedeco.cuda.cusparse.*;
+import static org.bytedeco.cuda.global.cusparse.*;
+import org.bytedeco.cuda.cusolver.*;
+import static org.bytedeco.cuda.global.cusolver.*;
 
 import static org.bytedeco.pytorch.global.torch_cuda.*;
 

diff --git a/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/CUDAKernelLaunchRegistry.java b/pytorch/src/gen/java/org/bytedeco/pytorch/cuda/CUDAKernelLaunchRegistry.java
@@ -4,11 +4,6 @@
 
 import org.bytedeco.pytorch.*;
 import org.bytedeco.pytorch.helper.*;
-import org.bytedeco.cuda.cudart.*;
-import org.bytedeco.cuda.cusparse.*;
-import org.bytedeco.cuda.cublas.*;
-import org.bytedeco.cuda.cusolver.*;
-import org.bytedeco.cuda.cudnn.*;
 import org.bytedeco.pytorch.chrono.*;
 import org.bytedeco.pytorch.global.torch.DeviceType;
 import org.bytedeco.pytorch.global.torch.ScalarType;
@@ -23,6 +18,16 @@
 import static org.bytedeco.openblas.global.openblas.*;
 import org.bytedeco.pytorch.*;
 import static org.bytedeco.pytorch.global.torch.*;
+import org.bytedeco.cuda.cudart.*;
+import static org.bytedeco.cuda.global.cudart.*;
+import org.bytedeco.cuda.cublas.*;
+import static org.bytedeco.cuda.global.cublas.*;
+import org.bytedeco.cuda.cudnn.*;
+import static org.bytedeco.cuda.global.cudnn.*;
+import org.bytedeco.cuda.cusparse.*;
+import static org.bytedeco.cuda.global.cusparse.*;
+import org.bytedeco.cuda.cusolver.*;
+import static org.bytedeco.cuda.global.cusolver.*;
 
 import static org.bytedeco.pytorch.global.torch_cuda.*;