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

Move TorchDynamo into PyTorch core (#86461)

2022-10-13 23:18:06 +00:00

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