Michael Lazos 9fa07340fd [Cutlass] Implement memory planning for EVT (#153177) ...

Pull Request resolved: https://github.com/pytorch/pytorch/pull/153177
Approved by: https://github.com/henrylhtsang
ghstack dependencies: #153196, #150907

2025-05-09 05:39:05 +00:00

..

aoti_runtime

[AOTInductor] Add User Managed buffer for AOTI constant buffer. (#150276)

2025-04-10 00:15:44 +00:00

cuda

[Cutlass] Implement memory planning for EVT (#153177)

2025-05-09 05:39:05 +00:00

rocm

Reland: [inductor] Simplify grid handling (#148305)

2025-03-12 15:52:16 +00:00

xpu

Reland: [inductor] Simplify grid handling (#148305)

2025-03-12 15:52:16 +00:00

__init__.py

Move TorchDynamo into PyTorch core (#86461)

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

aoti_hipify_utils.py

remove allow-untyped-defs from _inductor/codegen/aoti_hipify_utils.py (#143916)

2024-12-27 23:25:37 +00:00

block_analysis.py

[inductor][triton] Block ptr analysis fix assert on matched index expression (#148446)

2025-03-10 05:26:55 +00:00

common.py

[Cutlass] Implement memory planning for EVT (#153177)

2025-05-09 05:39:05 +00:00

cpp_bmm_template.py

[inductor][cpu] Move VNNI weight packing into AMX GEMM kernel for contiguous BMM weights (#146843)

2025-02-21 21:46:00 +00:00

cpp_flex_attention_template.py

Add pack support and use micro gemm for Half flex attention on CPU (#151530)

2025-04-29 07:24:00 +00:00

cpp_gemm_template.py

[Inductor][CPP] Optimize the epilogue for int8 GEMM Template (#152000)

2025-04-24 23:36:00 +00:00

cpp_grouped_gemm_template.py

[BE][PYFMT] migrate PYFMT for torch._inductor to ruff format (#144550)

2025-02-28 13:33:19 +00:00

cpp_micro_gemm.py

Revert "[AOTI][reland] Remove typedef for half and bfloat16 (#151109)"

2025-04-29 22:37:16 +00:00

cpp_prefix.h

Revert "[AOTI][reland] Remove typedef for half and bfloat16 (#151109)"

2025-04-29 22:37:16 +00:00

cpp_template_kernel.py

Reland: [inductor] Simplify grid handling (#148305)

2025-03-12 15:52:16 +00:00

cpp_template.py

[Inductor] Unify the data type propagation between Triton and CPP Backend (#146970)

2025-03-21 17:52:51 +00:00

cpp_utils.py

Revert "[AOTI][reland] Remove typedef for half and bfloat16 (#151109)"

2025-04-29 22:37:16 +00:00

cpp_wrapper_cpu_array_ref.py

[Inductor] Refactor wrapper codegen to use Wrapper IR. (#150458)

2025-04-15 17:28:36 +00:00

cpp_wrapper_cpu.py

[aoti] skip input symbol codegen for sympy expr w/ many symbols (#152579)

2025-05-07 01:18:09 +00:00

cpp_wrapper_gpu.py

[Inductor] Refactor wrapper codegen to use Wrapper IR. (#150458)

2025-04-15 17:28:36 +00:00

cpp.py

get right function declaration on windows inductor (#152939)

2025-05-08 14:28:33 +00:00

cpu_device_op_overrides.py

[inductor] Add types to DeviceOpOverrides (#145913)

2025-02-01 16:33:49 +00:00

cuda_combined_scheduling.py

[Cutlass] Integrate EVT into CUDACPPScheduling (#150906)

2025-05-07 23:09:02 +00:00

debug_utils.py

[Inductor] Refactor wrapper codegen to use Wrapper IR. (#150458)

2025-04-15 17:28:36 +00:00

halide.py

Remove libdevice ops in inductor (#151562)

2025-04-17 22:18:00 +00:00

memory_planning.py

[BE][PYFMT] migrate PYFMT for torch._inductor to ruff format (#144550)

2025-02-28 13:33:19 +00:00

mps_device_op_overrides.py

[MPSInductor] Fix noop codegen (#151224)

2025-04-14 16:38:47 +00:00

mps.py

[MPSInductor] Fix truncdiv implementation (#152788)

2025-05-05 13:31:51 +00:00

multi_kernel.py

[Inductor] Refactor wrapper codegen to use Wrapper IR. (#150458)

2025-04-15 17:28:36 +00:00

simd_kernel_features.py

[BE][Ez]: Use itertools.chain.from_iterable when possible (#148190)

2025-03-06 20:37:06 +00:00

simd.py

[Cutlass] Integrate EVT into CUDACPPScheduling (#150906)

2025-05-07 23:09:02 +00:00

subgraph.py

[Inductor] Add decomposeK as an autotuning choice for mm (#150654)

2025-05-03 02:23:54 +00:00

triton_combo_kernel.py

Reland: [inductor] Simplify grid handling (#148305)

2025-03-12 15:52:16 +00:00

triton_split_scan.py

Reland: [inductor] Simplify grid handling (#148305)

2025-03-12 15:52:16 +00:00

triton_utils.py

Fix broken URLs (#152237)

2025-04-27 09:56:42 +00:00

triton.py

Fix broken URLs (#152237)

2025-04-27 09:56:42 +00:00

wrapper_fxir.py

[Inductor] FX backend via Wrapper IR (#146942)

2025-05-06 10:06:39 +00:00

wrapper.py

[aoti] skip input symbol codegen for sympy expr w/ many symbols (#152579)

2025-05-07 01:18:09 +00:00