OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
06f619d999
pytorch/torch/_inductor/codegen
History
Aaron Orenstein 06f619d999 typing ir.py - part 2 (#131846) 
See #131852

Pull Request resolved: https://github.com/pytorch/pytorch/pull/131846
Approved by: https://github.com/eellison
ghstack dependencies: #139238
2024-11-06 00:01:15 +00:00
..
aoti_runtime [AOTI][refactor] Separate header codegen (#138882) 2024-10-27 14:14:27 +00:00
cuda typing ir.py - part 2 (#131846) 2024-11-06 00:01:15 +00:00
rocm typing ir.py - part 2 (#131846) 2024-11-06 00:01:15 +00:00
xpu [Inductor] Generalize device guard codegen for cpp_wrapper mode. (#134761) 2024-09-10 10:11:52 +00:00
__init__.py
aoti_hipify_utils.py [BE][Easy][16/19] enforce style for empty lines in import segments in torch/_i*/ (#129768) 2024-07-20 16:20:58 +00:00
common.py [inductor] sympy.Integer([01]) -> sympy.S.(Zero|One) (#139523) 2024-11-04 04:28:40 +00:00
cpp_gemm_template.py [Inductor][CPP] Add oneDNN BRGEMM config for Half cpp gemm template (#136255) 2024-11-05 05:33:29 +00:00
cpp_micro_gemm.py [Inductor][CPP] Add oneDNN BRGEMM config for Half cpp gemm template (#136255) 2024-11-05 05:33:29 +00:00
cpp_prefix.h std::value/std::type -> std::_v/std::_t (#138746) 2024-10-26 20:59:24 +00:00
cpp_template_kernel.py [inductor] sympy.Integer([01]) -> sympy.S.(Zero|One) (#139523) 2024-11-04 04:28:40 +00:00
cpp_template.py [AOTI] Remove the non-ABI-compatible mode (part 1) (#138009) 2024-10-17 02:48:26 +00:00
cpp_utils.py Add dtype attribute to CSEVariable (#136778) 2024-10-25 18:00:30 +00:00
cpp_wrapper_cpu_array_ref.py [AOTI][refactor] Move convert_arrayref_tensor_to_tensor logic (#139030) 2024-10-28 18:00:41 +00:00
cpp_wrapper_cpu.py [aoti] Print output name for sympy.Expr as well (#138524) 2024-10-29 16:02:45 +00:00
cpp_wrapper_gpu.py Add host-side TMA support to AOTInductor (#138878) 2024-10-28 23:39:53 +00:00
cpp.py typing ir.py - part 2 (#131846) 2024-11-06 00:01:15 +00:00
cpu_device_op_overrides.py Add Triton CPU as an Inductor backend (#133408) 2024-09-30 20:24:52 +00:00
cuda_combined_scheduling.py [BE]: Update mypy to 1.11.2 (#133816) 2024-09-16 19:44:11 +00:00
debug_utils.py [AOTI] Remove the non-ABI-compatible mode (part 1) (#138009) 2024-10-17 02:48:26 +00:00
halide.py [inductor] sympy.Integer([01]) -> sympy.S.(Zero|One) (#139523) 2024-11-04 04:28:40 +00:00
memory_planning.py [inductor] Generalize WorkspaceArg for graph-level semaphores (#138170) 2024-10-18 23:05:54 +00:00
multi_kernel.py [inductor] Multi-kernel + cooperative reductions (#138893) 2024-10-29 15:45:17 +00:00
simd_kernel_features.py [inductor] sympy.Integer([01]) -> sympy.S.(Zero|One) (#139523) 2024-11-04 04:28:40 +00:00
simd.py [inductor] sympy.Integer([01]) -> sympy.S.(Zero|One) (#139523) 2024-11-04 04:28:40 +00:00
triton_combo_kernel.py [inductor] Refactor kernel args into SIMDKernelFeatures (#139327) 2024-11-01 00:30:14 +00:00
triton_split_scan.py [inductor] Refactor kernel args into SIMDKernelFeatures (#139327) 2024-11-01 00:30:14 +00:00
triton_utils.py [inductor] Move V.graph.scheduler.current_device to V.graph.current_device (#138252) 2024-10-18 23:05:54 +00:00
triton.py [inductor] sympy.Integer([01]) -> sympy.S.(Zero|One) (#139523) 2024-11-04 04:28:40 +00:00
wrapper.py Revert "Allow inplacing buffer when other users are inconsequential (#138383)" 2024-11-02 06:53:48 +00:00