pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

Isuru Fernando 99b32a6750 [inductor] print 0.0 as 0 for triton (#164291 )

Fixes https://github.com/pytorch/pytorch/issues/164157
Fixes https://github.com/pytorch/pytorch/issues/164086

Pull Request resolved: https://github.com/pytorch/pytorch/pull/164291
Approved by: https://github.com/bobrenjc93

2025-10-16 16:37:50 +00:00

analysis

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

autoheuristic

Update ruff to 0.13.1 (#163744 )

2025-09-26 10:12:21 +00:00

codegen

[inductor] print 0.0 as 0 for triton (#164291 )

2025-10-16 16:37:50 +00:00

compile_worker

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

fx_passes

Fx collectives bucketing: add bucket all_reduce (#165351 )

2025-10-16 13:27:33 +00:00

kernel

[Inductor][CuTeDSL] Move load_template up two directories (#165347 ) (#165576 )

2025-10-15 23:37:55 +00:00

package

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

runtime

Codemod inductor/runtime from Optional to union none (#165605 )

2025-10-16 04:59:47 +00:00

template_heuristics

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

__autotune_main__.py

Improve subproc autotuning implementation (#149700 )

2025-03-28 01:06:39 +00:00

__init__.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

analyze_preserves_zero_mask.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

aoti_eager.py

PEP585 update - torch/_inductor/[_-i]* (#145137 )

2025-01-19 01:22:47 +00:00

async_compile.py

[multi-kernel] shape-similarity kernel selection (#163090 )

2025-09-23 21:00:47 +00:00

augmented_graph_helper.py

Limit coll bucketing within node idxs (#164944 )

2025-10-08 22:18:53 +00:00

autotune_process.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

await_utils.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

bounds.py

[inductor] Refactor op handlers part 5 (#146257 )

2025-02-08 18:00:30 +00:00

cache.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

choices.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

codecache.py

Codemod codecache.py from Optional to union none (#165604 )

2025-10-16 04:59:37 +00:00

comm_analysis.py

Overlap scheduler improvements (#165318 )

2025-10-15 21:58:47 +00:00

comm_lowering.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

comms_debug.py

[inductor] Estimate peak memory allocfree and applying to reordering collectives (#160113 )

2025-08-22 14:19:57 +00:00

comms.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

compile_fx_async.py

[pc] introduce ProgressiveCompilationState and clear callback (#157619 )

2025-07-05 07:55:11 +00:00

compile_fx_ext.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

compile_fx_subproc.py

Rename inductor cache (#156128 )

2025-06-17 03:57:18 +00:00

compile_fx.py

[inductor] verify determinism with inductor benchmark script (#164904 )

2025-10-12 00:03:42 +00:00

compiler_bisector.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

config_comms.py

[inductor] Runtime estimations: use nccl estimator; mm only benchmark mode (#161405 )

2025-09-08 14:33:19 +00:00

config.py

Gate division bitwise numerics under a flag (#165566 )

2025-10-15 23:41:01 +00:00

constant_folding.py

[2/N] Use "is" in python type comparison (#165142 )

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

cpp_builder.py

Support libtorch and posix mingw flavor (#165574 )

2025-10-16 01:31:50 +00:00

cpu_vec_isa.py

Revert "Add SVE128 ISA (#158932 )"

2025-10-10 01:17:02 +00:00

cudagraph_trees.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

cudagraph_utils.py

[CUDAGraph] add config to error on skipping cudagraph (#161862 )

2025-09-04 15:52:39 +00:00

custom_graph_pass.py

[inductor] custom_graph_pass.get_hash_for_files: don't hash paths (#165020 )

2025-10-09 20:07:53 +00:00

debug.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

decomposition.py

[Inductor] Fix out-of-bounds indices in repeat_interleave decomposition (#165368 )

2025-10-14 21:16:36 +00:00

dependencies.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

dtype_propagation.py

Native matmul (#157743 )

2025-10-14 04:22:30 +00:00

exc.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

extern_node_serializer.py

[BE][AOTI] Remove duplicate schema for ExternKernelNode (#155867 )

2025-06-14 02:03:27 +00:00

freezing_utils.py

PEP585: More UP006 fixes (#146392 )

2025-02-20 06:18:13 +00:00

freezing.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

fuzzer.py

Enable ruff rule E721 (#165162 )

2025-10-13 01:48:55 +00:00

fx_utils.py

[1/N] Use "is" in python type comparison (#165037 )

2025-10-10 12:36:50 +00:00

graph.py

[Graph Partition] fix partition x memory plan issue (#165514 )

2025-10-15 21:52:16 +00:00

hooks.py

PEP585 update - torch/_inductor/[_-i]* (#145137 )

2025-01-19 01:22:47 +00:00

index_propagation.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

inductor_prims.py

[Easy] Add pointwise tag to fma (#164149 )

2025-09-29 22:40:04 +00:00

ir.py

Revert "[inductor] Expand use of generic benchmark function (#164938 )"

2025-10-15 17:48:38 +00:00

jagged_lowerings.py

Fix XPU CI UT test_circular_dependencies (#158189 )

2025-07-13 09:30:57 +00:00

kernel_inputs.py

[inductor][template heuristics] don't take layout to generate choices (#162238 )

2025-09-09 17:17:04 +00:00

kernel_template_choice.py

[inductor][choices] move extra kwargs out of get_template_configs (#163209 )

2025-09-20 05:30:40 +00:00

loop_body.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

lowering.py

[Bugfix][Dynamo] Fix Sparse tensors by graph break in Dynamo (#164873 )

2025-10-16 15:06:20 +00:00

memory.py

An improved heuristic for operator reordering for peak memory + debugging logs (#161810 )

2025-09-13 00:42:32 +00:00

metrics.py

[inductor] log kernel autotuning result to a csv (#164191 )

2025-10-02 00:25:34 +00:00

mkldnn_ir.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

mkldnn_lowerings.py

[2/N] Use "is" in python type comparison (#165142 )

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

mock_cache.py

PEP585 update - torch/_inductor (#145198 )

2025-01-21 21:04:33 +00:00

ops_handler.py

Native matmul (#157743 )

2025-10-14 04:22:30 +00:00

optimize_indexing.py

PEP585 update - torch/_inductor (#145198 )

2025-01-21 21:04:33 +00:00

output_code.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

pattern_matcher.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

quantized_lowerings.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

remote_cache.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

remote_gemm_autotune_cache.py

integrate kernacle into inductor (#160121 )

2025-08-08 02:14:44 +00:00

scheduler.py

Revert "[inductor] Expand use of generic benchmark function (#164938 )"

2025-10-15 17:48:38 +00:00

script.ld

Place .lrodata later in the binary (#117575 )

2024-01-18 17:58:18 +00:00

select_algorithm.py

Revert "[inductor] Expand use of generic benchmark function (#164938 )"

2025-10-15 17:48:38 +00:00

shape_propagation.py

Native matmul (#157743 )

2025-10-14 04:22:30 +00:00

sizevars.py

[inductor] fix issue for example value with unbacked strides (#163660 )

2025-10-14 20:07:51 +00:00

standalone_compile.py

[standalone_compile] binary format write should be atomic (#162432 )

2025-09-09 18:43:13 +00:00

subgraph_lowering.py

[FlexAttention] fixing learnable bias assertion error in inductor (#161170 )

2025-08-23 06:24:22 +00:00

test_case.py

Rename inductor cache (#156128 )

2025-06-17 03:57:18 +00:00

test_operators.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

tiling_utils.py

Add suppressions to torch/_inductor (#165062 )

2025-10-09 20:34:20 +00:00

triton_bundler.py

[inductor][triton] more JITCallable._hash_lock support (#162244 )

2025-09-08 07:57:48 +00:00

utils.py

[Inductor][CuTeDSL] Move load_template up two directories (#165347 ) (#165576 )

2025-10-15 23:37:55 +00:00

virtualized.py

[export] Fix custom ops in subgraphs (#160004 )

2025-08-18 15:42:19 +00:00

wrapper_benchmark.py

Revert "[inductor] Expand use of generic benchmark function (#164938 )"

2025-10-15 17:48:38 +00:00