OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
cc8f1cddd4
pytorch/torch/_inductor/codegen
History
Edward Z. Yang cc8f1cddd4 Turn on type-checking in torch.fx.experimental.symbolic_shapes (#136972) 
Signed-off-by: Edward Z. Yang <ezyang@meta.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/136972
Approved by: https://github.com/Skylion007
ghstack dependencies: #136934, #136935
2024-10-01 13:22:10 +00:00
..
aoti_runtime [aoti][inplace] Support skipping model buffers (#136770) 2024-09-30 18:28:42 +00:00
cuda [Inductor] Rename WrapperCodeGen to PythonWrapperCodegen (#136062) 2024-09-24 21:02:51 +00:00
rocm Add Triton CPU as an Inductor backend (#133408) 2024-09-30 20:24:52 +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 Add Triton CPU as an Inductor backend (#133408) 2024-09-30 20:24:52 +00:00
cpp_gemm_template.py [inductor] [cpp] fix gemm_output_name conflict (#136419) 2024-09-27 12:23:17 +00:00
cpp_micro_gemm.py [Inductor][CPP] Optimize WOQ INT8 wgt dequant in AMX GEMM template (#136630) 2024-09-26 08:41:58 +00:00
cpp_prefix.h [Inductor] Further solve missing aoti_torch_check symbole issue (#136775) 2024-09-27 02:26:49 +00:00
cpp_template_kernel.py Add Triton CPU as an Inductor backend (#133408) 2024-09-30 20:24:52 +00:00
cpp_template.py [AOTI][reland] Fix assert_function call in cpu autotune template (#135920) 2024-09-13 12:21:57 +00:00
cpp_utils.py [inductor] [cpp] add index check when fusing epilogue with GEMM template (#135661) 2024-09-24 05:25:28 +00:00
cpp_wrapper_cpu_array_ref.py Fix AOT Graph capture not propagating non_blocking copy parameter to … (#136513) 2024-10-01 00:32:47 +00:00
cpp_wrapper_cpu.py Fix AOT Graph capture not propagating non_blocking copy parameter to … (#136513) 2024-10-01 00:32:47 +00:00
cpp_wrapper_gpu.py [aoti] Fix workspace generation for triton (#135552) 2024-09-22 04:51:37 +00:00
cpp.py Add Triton CPU as an Inductor backend (#133408) 2024-09-30 20:24:52 +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][Tooling][9/n] Add debug printer support for cpp kernel type (#136465) 2024-09-26 17:30:43 +00:00
halide.py Add Triton CPU as an Inductor backend (#133408) 2024-09-30 20:24:52 +00:00
memory_planning.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
multi_kernel.py remove fast_flush arguments (#135387) 2024-09-13 08:13:46 +00:00
simd.py [Inductor] Remove stride-0 dimensions from more complex block pointers (#135557) 2024-09-27 04:01:40 +00:00
triton_combo_kernel.py [pytorch][PR] [inductor] More fixes on the keys of constants and signature dictionaries (#136514) 2024-09-25 09:34:14 +00:00
triton_split_scan.py [aoti] Fix workspace generation for triton (#135552) 2024-09-22 04:51:37 +00:00
triton_utils.py [pytorch][PR] [inductor] More fixes on the keys of constants and signature dictionaries (#136514) 2024-09-25 09:34:14 +00:00
triton.py Add Triton CPU as an Inductor backend (#133408) 2024-09-30 20:24:52 +00:00
wrapper.py Turn on type-checking in torch.fx.experimental.symbolic_shapes (#136972) 2024-10-01 13:22:10 +00:00