OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
39df901b2a
pytorch/torch/_inductor/codegen
History
Laith Sakka 39df901b2a introduce definitely_contiguous and use it for reshape and tensor meta data computation. (#153432) 
when a tensor has unbacked symbols it can be general enough to represent both contiguous and non contiguous tensors.
in that case we cant really evaluate is_contiguous. In many places in the code base, we check for is_contiguous to take a fast path. but the general path usually works for both contiguous and not contiguous in that case we probably want
to use definitely _contiguous API.

This is appleid for reshape in this PR and also to  tensor meta data computation, the meta data now will have an attribute that says that its contiguous when its always contiguous. We would store that only if definitely _contiguous is true  now.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/153432
Approved by: https://github.com/bobrenjc93
2025-05-28 03:41:26 +00:00
..
aoti_runtime [AOTInductor] Fix clang-tidy warnings in wrapper (#153197) 2025-05-12 22:35:59 +00:00
cuda Update to using mypy 1.15 (#154054) 2025-05-24 04:30:57 +00:00
rocm [ROCm][Inductor][CK] Add ck-tile based universal gemm kernels to torch.mm autotune choices (#152341) 2025-05-21 23:59:16 +00:00
xpu Reland: [inductor] Simplify grid handling (#148305) 2025-03-12 15:52:16 +00:00
__init__.py
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 [aoti] Initial Metal support (#153959) 2025-05-23 05:45:35 +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-CPU] Faster int8 WoQ GEMM for small M with explicit prefetching and different outer loops (#149373) 2025-05-15 11:55:58 +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 [Inductor-CPU] Faster int8 WoQ GEMM for small M with explicit prefetching and different outer loops (#149373) 2025-05-15 11:55:58 +00:00
cpp_template_kernel.py [Inductor-CPU] Faster int8 WoQ GEMM for small M with explicit prefetching and different outer loops (#149373) 2025-05-15 11:55:58 +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 Revert "[Inductor] Improve typing, and prepare for ABI-compatible AOTI C-shim dispatching (#154371)" 2025-05-27 20:39:09 +00:00
cpp_wrapper_cpu.py [AOTI][refactor] Rename embed_cubin to embed_kernel_binary (#154412) 2025-05-28 01:20:28 +00:00
cpp_wrapper_gpu.py [AOTI][refactor] Rename embed_cubin to embed_kernel_binary (#154412) 2025-05-28 01:20:28 +00:00
cpp_wrapper_mps.py [aoti] Initial Metal support (#153959) 2025-05-23 05:45:35 +00:00
cpp.py [Inductor][CPP] Enable vectorized fp8 E5M2 quant dequant (#153365) 2025-05-23 23:20:02 +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 [BE]: Update ruff to 0.11.8 (#153249) 2025-05-12 18:30:52 +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 [aoti] Initial Metal support (#153959) 2025-05-23 05:45:35 +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 introduce definitely_contiguous and use it for reshape and tensor meta data computation. (#153432) 2025-05-28 03:41:26 +00:00
subgraph.py [Inductor] Subgraph check output strides (#153755) 2025-05-20 16:07:18 +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] Optimize grid calculation by using // instead of FloorDiv (#153230) 2025-05-12 18:08:52 +00:00
wrapper.py Revert "[Inductor] Improve typing, and prepare for ABI-compatible AOTI C-shim dispatching (#154371)" 2025-05-27 20:39:09 +00:00