pytorch/torch/csrc/inductor
Bin Bao 3058700f7f [aotinductor] Add AOTIModelRunner as a utility class (#110891)
Summary: Introduce a utility class AOTIModelRunner to take care of running an AOTInductor compiled model. It does things like dlopen a model, initialize the model container, setup inputs and outputs, and destroy the model container.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/110891
Approved by: https://github.com/chenyang78
ghstack dependencies: #110652
2023-10-11 15:58:28 +00:00
..
aoti_runtime [inductor] Add size, stride, storage_offset to RAIIAtenTensorHandle (#110764) 2023-10-07 08:26:42 +00:00
aoti_torch [inductor] Add AOTI ABI shim function for torch.nonzero (#110766) 2023-10-07 08:32:27 +00:00
aoti_model_runner_cuda.cpp [aotinductor] Add AOTIModelRunner as a utility class (#110891) 2023-10-11 15:58:28 +00:00
aoti_model_runner_cuda.h [aotinductor] Add AOTIModelRunner as a utility class (#110891) 2023-10-11 15:58:28 +00:00
aoti_model_runner.cpp [aotinductor] Add AOTIModelRunner as a utility class (#110891) 2023-10-11 15:58:28 +00:00
aoti_model_runner.h [aotinductor] Add AOTIModelRunner as a utility class (#110891) 2023-10-11 15:58:28 +00:00
inductor_ops.cpp [inductor] Fix inputs with existing offsets (#108168) 2023-08-29 23:47:03 +00:00
inductor_ops.h [inductor] Fix inputs with existing offsets (#108168) 2023-08-29 23:47:03 +00:00