frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

[aotinductor] Rename aot_runtime to aoti_runtime (#110007)

Browse Source 
Summary: Make the naming more explicit

Differential Revision: D49593528

Pull Request resolved: https://github.com/pytorch/pytorch/pull/110007
Approved by: https://github.com/houseroad
This commit is contained in:
Bin Bao
2023-09-26 00:46:54 +00:00
committed by PyTorch MergeBot
parent b07bebd4bd
commit 4bf1cd6961
9 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
View File

@ -1293,7 +1293,7 @@ def main():
"include/torch/csrc/distributed/autograd/rpc_messages/*.h",
"include/torch/csrc/dynamo/*.h",
"include/torch/csrc/inductor/*.h",
"include/torch/csrc/inductor/aot_runtime/*.h",
"include/torch/csrc/inductor/aoti_runtime/*.h",
"include/torch/csrc/inductor/aoti_torch/*.h",
"include/torch/csrc/inductor/aoti_torch/c/*.h",
"include/torch/csrc/jit/*.h",
@ -1339,7 +1339,7 @@ def main():
"include/THH/generic/*.h",
"include/sleef.h",
"_inductor/codegen/*.h",
"_inductor/codegen/aot_runtime/*.cpp",
"_inductor/codegen/aoti_runtime/*.cpp",
"share/cmake/ATen/*.cmake",
"share/cmake/Caffe2/*.cmake",
"share/cmake/Caffe2/public/*.cmake",

2
test/cpp/aot_inductor/test.cpp
View File

@ -3,7 +3,7 @@
#include <vector>
#include <c10/cuda/CUDAStream.h>
#include <torch/csrc/inductor/aot_runtime/interface.h>
#include <torch/csrc/inductor/aoti_runtime/interface.h>
#include <torch/csrc/inductor/aoti_torch/tensor_converter.h>
#include <torch/torch.h>

4
torch/_inductor/codegen/aot_runtime/interface.cpp → torch/_inductor/codegen/aoti_runtime/interface.cpp
View File

@ -1,5 +1,5 @@
#include <torch/csrc/inductor/aot_runtime/interface.h>
#include <torch/csrc/inductor/aot_runtime/model_container.h>
#include <torch/csrc/inductor/aoti_runtime/interface.h>
#include <torch/csrc/inductor/aoti_runtime/model_container.h>
#include <iostream>
#include <stdexcept>

2
torch/_inductor/codegen/wrapper.py
View File

@ -1019,7 +1019,7 @@ class CppWrapperCodeGen(WrapperCodeGen):
def write_header(self):
if V.graph.aot_mode:
with open(
os.path.join(os.path.dirname(__file__), "aot_runtime", "interface.cpp")
os.path.join(os.path.dirname(__file__), "aoti_runtime", "interface.cpp")
) as f:
self.header.splice(f.read())
else:

2
torch/_inductor/utils.py
View File

@ -1175,7 +1175,7 @@ class Placeholder(enum.Enum):
# A utility function for easier AOTInductor testing
aot_inductor_launcher = """
#include <c10/cuda/CUDAStream.h>
#include <torch/csrc/inductor/aot_runtime/interface.h>
#include <torch/csrc/inductor/aoti_runtime/interface.h>
#include <torch/csrc/inductor/aoti_torch/tensor_converter.h>
class RAIIModelContainer {

2
torch/csrc/inductor/aot_runtime/cuda_utils.h → torch/csrc/inductor/aoti_runtime/cuda_utils.h
View File

@ -3,7 +3,7 @@
// WARNING: Be careful when adding new includes here. This header will be used
// in model.so, and should not refer to any aten/c10 headers except the stable
// C ABI defined in torch/csrc/inductor/aoti_torch/c/shim.h. The same rule
// applies to other files under torch/csrc/inductor/aot_runtime/.
// applies to other files under torch/csrc/inductor/aoti_runtime/.
#include <cuda.h>
#include <cuda_runtime_api.h>

2
torch/csrc/inductor/aot_runtime/interface.h → torch/csrc/inductor/aoti_runtime/interface.h
View File

@ -6,7 +6,7 @@
// WARNING: Be careful when adding new includes here. This header will be used
// in model.so, and should not refer to any aten/c10 headers except the stable
// C ABI defined in torch/csrc/inductor/aoti_torch/c/shim.h. The same rule
// applies to other files under torch/csrc/inductor/aot_runtime/.
// applies to other files under torch/csrc/inductor/aoti_runtime/.
#include <torch/csrc/inductor/aoti_torch/c/shim.h>
#ifdef __GNUC__

4
torch/csrc/inductor/aot_runtime/model.h → torch/csrc/inductor/aoti_runtime/model.h
View File

@ -10,8 +10,8 @@
// WARNING: Be careful when adding new includes here. This header will be used
// in model.so, and should not refer to any aten/c10 headers except the stable
// C ABI defined in torch/csrc/inductor/aoti_torch/c/shim.h. The same rule
// applies to other files under torch/csrc/inductor/aot_runtime/.
#include <torch/csrc/inductor/aot_runtime/cuda_utils.h>
// applies to other files under torch/csrc/inductor/aoti_runtime/.
#include <torch/csrc/inductor/aoti_runtime/cuda_utils.h>
#include <torch/csrc/inductor/aoti_torch/c/shim.h>
#define AOTI_RUNTIME_CHECK(EXPR, MSG) \

4
torch/csrc/inductor/aot_runtime/model_container.h → torch/csrc/inductor/aoti_runtime/model_container.h
View File

@ -9,8 +9,8 @@
// WARNING: Be careful when adding new includes here. This header will be used
// in model.so, and should not refer to any aten/c10 headers except the stable
// C ABI defined in torch/csrc/inductor/aoti_torch/c/shim.h. The same rule
// applies to other files under torch/csrc/inductor/aot_runtime/.
#include <torch/csrc/inductor/aot_runtime/model.h>
// applies to other files under torch/csrc/inductor/aoti_runtime/.
#include <torch/csrc/inductor/aoti_runtime/model.h>
// At codegen time, we write out a binary file called constants.bin.
// We then turn the raw binary to an object file that exposes this