pytorch/torch/csrc/inductor/aoti_runtime
Adnan Akhundov 13a2f42635 [inductor] Add size, stride, storage_offset to RAIIAtenTensorHandle (#110764)
Summary: For unbacked SymInts, the C++ wrapper codegen can generate expressions like `buf123.size()` or `.stride()` or `.storage_offset()`:

7cc0020a80/torch/_inductor/ir.py (L2504-L2520)

Here we add corresponding methods to the `RAIIAtenTensorHandle` class so that the above codegen works in the ABI compatibility mode.

Test Plan: CI + the following PR

Reviewers:

Subscribers:

Tasks:

Tags:

Pull Request resolved: https://github.com/pytorch/pytorch/pull/110764
Approved by: https://github.com/chenyang78
ghstack dependencies: #110713, #110745
2023-10-07 08:26:42 +00:00
..
device_utils.h [AOTInductor] remove CUDA dependency for cpp backend (#110409) 2023-10-03 18:36:00 +00:00
interface.h [AOTInductor] Change UpdateConstants to UpdateConstantsMap (#110576) 2023-10-07 07:36:57 +00:00
model_container.h [AOTInductor] Simplified AOTInductor interface and model class (#110411) 2023-10-04 18:35:24 +00:00
model.h [inductor] Add size, stride, storage_offset to RAIIAtenTensorHandle (#110764) 2023-10-07 08:26:42 +00:00