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
26f7ca3972
pytorch/torch/_inductor/codegen
History
Laith Sakka 26f7ca3972 Unify dynamic shapes APIs naming 2 (expect_true and check) attempt2 (#156518) 
Summary:
The functions guard_lt, guard_equals, and guard_leq work similarly to torch.check and expect_true, but they operate on SymPy expressions. Notably, guard_equals applies local replacements before comparison, which might be better extracted into a separate function.

This pull request standardizes naming conventions to match symbolic_shapes.py. Specifically,
-  it introduces size_vars.expect_true and size_vars.check.
- guard_lt becomes check_lt
- guard_leq becomes check_leq
- guard_equals becomes check_equals

I am also seeing a couple of wrong usages !! that i will fix  in the next PR

Test Plan:
OSS and cont

Rollback Plan:

Differential Revision: D77054177

Pull Request resolved: https://github.com/pytorch/pytorch/pull/156518
Approved by: https://github.com/bobrenjc93
2025-06-24 21:01:38 +00:00
..
aoti_runtime [AOTI] Save data sizes to constants_info (#154534) 2025-05-29 06:39:13 +00:00
cuda [cutlass backend] delete pip cutlass path since nvidia stops supporting nvidia-cutlass (#156651) 2025-06-24 18:32:15 +00:00
rocm Migrate from lru_cache to cache (#155613) 2025-06-11 19:44:18 +00:00
xpu Reland: [inductor] Simplify grid handling (#148305) 2025-03-12 15:52:16 +00:00
__init__.py
aoti_hipify_utils.py [BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313) 2025-06-23 02:57:12 +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 [BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313) 2025-06-23 02:57:12 +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 [BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313) 2025-06-23 02:57:12 +00:00
cpp_gemm_template.py [BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313) 2025-06-23 02:57:12 +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 [BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313) 2025-06-23 02:57:12 +00:00
cpp_template_kernel.py Revert "[inductor] Add typing to _inductor/ir.py (#149958)" 2025-06-06 15:19:16 +00:00
cpp_template.py codecache: Remove cpp_prefix.h duplication per build, then precompile it (#144293) 2025-05-16 17:41:36 +00:00
cpp_utils.py [aoti] Initial Metal support (#153959) 2025-05-23 05:45:35 +00:00
cpp_wrapper_cpu_array_ref.py [BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313) 2025-06-23 02:57:12 +00:00
cpp_wrapper_cpu.py [aoti] Check longlong upperbound for codegening input size check (#156522) 2025-06-23 20:38:34 +00:00
cpp_wrapper_gpu.py [BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313) 2025-06-23 02:57:12 +00:00
cpp_wrapper_mps.py [aoti][mps] Fix int/symint kernel args (#155583) 2025-06-12 23:33:28 +00:00
cpp.py Unify dynamic shapes APIs naming 2 (expect_true and check) attempt2 (#156518) 2025-06-24 21:01:38 +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] E2E Tests for EVT (#152815) 2025-05-17 12:29:10 +00:00
debug_utils.py [Inductor] Refactor wrapper codegen to use Wrapper IR. (#150458) 2025-04-15 17:28:36 +00:00
halide.py Unify dynamic shapes APIs naming 2 (expect_true and check) attempt2 (#156518) 2025-06-24 21:01:38 +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 [aoti] Initial Metal support (#153959) 2025-05-23 05:45:35 +00:00
mps.py [MPSInductor][BE] Fix multistage reduction check (#156567) 2025-06-23 14:49:26 +00:00
multi_kernel.py [ez] remove unused functions (#156466) 2025-06-21 00:38:34 +00:00
simd_kernel_features.py Replace runtime type parameterization (#155221) 2025-06-05 21:43:54 +00:00
simd.py Unify dynamic shapes APIs naming 2 (expect_true and check) attempt2 (#156518) 2025-06-24 21:01:38 +00:00
subgraph.py [Inductor] Subgraph as a choice symbolic expression as input (#156185) 2025-06-23 21:29:17 +00:00
triton_combo_kernel.py [BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313) 2025-06-23 02:57:12 +00:00
triton_split_scan.py Reland: [inductor] Simplify grid handling (#148305) 2025-03-12 15:52:16 +00:00
triton_utils.py Unify dynamic shapes APIs naming 2 (expect_true and check) attempt2 (#156518) 2025-06-24 21:01:38 +00:00
triton.py [BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313) 2025-06-23 02:57:12 +00:00
wrapper_fxir.py [Inductor] Support autotuning in the FX backend. (#155049) 2025-06-06 17:44:14 +00:00
wrapper.py Remove unused MultiKernelCall import from inductor codegen (#156158) 2025-06-20 19:55:24 +00:00