-
Notifications
You must be signed in to change notification settings - Fork 751
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* Upgrade presets for PyTorch 2.3.0 (pull #1498)
- Loading branch information
1 parent
34289d4
commit f8932a4
Showing
104 changed files
with
2,878 additions
and
2,385 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
48 changes: 48 additions & 0 deletions
48
pytorch/src/gen/java/org/bytedeco/pytorch/AOTIModelContainerRunner.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,48 @@ | ||
| // Targeted by JavaCPP version 1.5.11-SNAPSHOT: DO NOT EDIT THIS FILE | ||
|
|
||
| package org.bytedeco.pytorch; | ||
|
|
||
| import org.bytedeco.pytorch.Allocator; | ||
| import org.bytedeco.pytorch.Function; | ||
| import org.bytedeco.pytorch.functions.*; | ||
| import org.bytedeco.pytorch.Module; | ||
| import org.bytedeco.javacpp.annotation.Cast; | ||
| import java.nio.*; | ||
| import org.bytedeco.javacpp.*; | ||
| import org.bytedeco.javacpp.annotation.*; | ||
|
|
||
| import static org.bytedeco.javacpp.presets.javacpp.*; | ||
| import static org.bytedeco.openblas.global.openblas_nolapack.*; | ||
| import static org.bytedeco.openblas.global.openblas.*; | ||
|
|
||
| import static org.bytedeco.pytorch.global.torch.*; | ||
|
|
||
|
|
||
| @Namespace("torch::inductor") @NoOffset @Properties(inherit = org.bytedeco.pytorch.presets.torch.class) | ||
| public class AOTIModelContainerRunner extends Pointer { | ||
| static { Loader.load(); } | ||
| /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ | ||
| public AOTIModelContainerRunner(Pointer p) { super(p); } | ||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
| public native @ByVal TensorVector run( | ||
| @ByRef TensorVector inputs); | ||
|
|
||
| public native @ByVal ExtraFilesMap getConstantNamesToOriginalFQNs(); | ||
| public native @ByVal StringIntMap getConstantNamesToDtypes(); | ||
| public native void update_inactive_constant_buffer(@Cast("const torch::inductor::TensorConstantMap*") @ByRef HashAliasedIValueMap const_map); | ||
| public native void update_constant_buffer( | ||
| @Cast("const torch::inductor::TensorConstantMap*") @ByRef HashAliasedIValueMap const_map, | ||
| @Cast("bool") boolean use_inactive, | ||
| @Cast("bool") boolean validate_full_updates); | ||
| public native void run_const_fold( | ||
| @Cast("bool") boolean use_inactive); | ||
| public native void swap_constant_buffer(); | ||
|
|
||
| public native @ByVal StringVector get_call_spec(); | ||
| } |
48 changes: 48 additions & 0 deletions
48
pytorch/src/gen/java/org/bytedeco/pytorch/AOTIModelContainerRunnerCpu.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,48 @@ | ||
| // Targeted by JavaCPP version 1.5.11-SNAPSHOT: DO NOT EDIT THIS FILE | ||
|
|
||
| package org.bytedeco.pytorch; | ||
|
|
||
| import org.bytedeco.pytorch.Allocator; | ||
| import org.bytedeco.pytorch.Function; | ||
| import org.bytedeco.pytorch.functions.*; | ||
| import org.bytedeco.pytorch.Module; | ||
| import org.bytedeco.javacpp.annotation.Cast; | ||
| import java.nio.*; | ||
| import org.bytedeco.javacpp.*; | ||
| import org.bytedeco.javacpp.annotation.*; | ||
|
|
||
| import static org.bytedeco.javacpp.presets.javacpp.*; | ||
| import static org.bytedeco.openblas.global.openblas_nolapack.*; | ||
| import static org.bytedeco.openblas.global.openblas.*; | ||
|
|
||
| import static org.bytedeco.pytorch.global.torch.*; | ||
|
|
||
| @Namespace("torch::inductor") @Properties(inherit = org.bytedeco.pytorch.presets.torch.class) | ||
| public class AOTIModelContainerRunnerCpu extends AOTIModelContainerRunner { | ||
| static { Loader.load(); } | ||
| /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ | ||
| public AOTIModelContainerRunnerCpu(Pointer p) { super(p); } | ||
|
|
||
| public AOTIModelContainerRunnerCpu( | ||
| @StdString BytePointer model_so_path, | ||
| @Cast("size_t") long num_models/*=1*/) { super((Pointer)null); allocate(model_so_path, num_models); } | ||
| private native void allocate( | ||
| @StdString BytePointer model_so_path, | ||
| @Cast("size_t") long num_models/*=1*/); | ||
| public AOTIModelContainerRunnerCpu( | ||
| @StdString BytePointer model_so_path) { super((Pointer)null); allocate(model_so_path); } | ||
| private native void allocate( | ||
| @StdString BytePointer model_so_path); | ||
| public AOTIModelContainerRunnerCpu( | ||
| @StdString String model_so_path, | ||
| @Cast("size_t") long num_models/*=1*/) { super((Pointer)null); allocate(model_so_path, num_models); } | ||
| private native void allocate( | ||
| @StdString String model_so_path, | ||
| @Cast("size_t") long num_models/*=1*/); | ||
| public AOTIModelContainerRunnerCpu( | ||
| @StdString String model_so_path) { super((Pointer)null); allocate(model_so_path); } | ||
| private native void allocate( | ||
| @StdString String model_so_path); | ||
|
|
||
| public native @ByVal TensorVector run(@ByRef TensorVector inputs); | ||
| } |
37 changes: 37 additions & 0 deletions
37
pytorch/src/gen/java/org/bytedeco/pytorch/AcceleratorHooksInterface.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,37 @@ | ||
| // Targeted by JavaCPP version 1.5.11-SNAPSHOT: DO NOT EDIT THIS FILE | ||
|
|
||
| package org.bytedeco.pytorch; | ||
|
|
||
| import org.bytedeco.pytorch.Allocator; | ||
| import org.bytedeco.pytorch.Function; | ||
| import org.bytedeco.pytorch.functions.*; | ||
| import org.bytedeco.pytorch.Module; | ||
| import org.bytedeco.javacpp.annotation.Cast; | ||
| import java.nio.*; | ||
| import org.bytedeco.javacpp.*; | ||
| import org.bytedeco.javacpp.annotation.*; | ||
|
|
||
| import static org.bytedeco.javacpp.presets.javacpp.*; | ||
| import static org.bytedeco.openblas.global.openblas_nolapack.*; | ||
| import static org.bytedeco.openblas.global.openblas.*; | ||
|
|
||
| import static org.bytedeco.pytorch.global.torch.*; | ||
|
|
||
|
|
||
| // AcceleratorHooksInterface is a shared interface provided by all | ||
| // accelerators to allow generic code. | ||
| // This inferface is hook-based as it corresponds to all the functions | ||
| // that are going to be called in a generic way from the CPU code. | ||
|
|
||
| @Namespace("at") @Properties(inherit = org.bytedeco.pytorch.presets.torch.class) | ||
| public class AcceleratorHooksInterface extends Pointer { | ||
| static { Loader.load(); } | ||
| /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ | ||
| public AcceleratorHooksInterface(Pointer p) { super(p); } | ||
|
|
||
| // This should never actually be implemented, but it is used to | ||
| // squelch -Werror=non-virtual-dtor | ||
|
|
||
| // Whether the device at device_index is fully initialized or not. | ||
| public native @Cast("bool") boolean hasPrimaryContext(@Cast("c10::DeviceIndex") byte device_index); | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.