pytorch

cpp

…

extension_backends

Make Context to be Device-agnostic Step by Step (3/N) (#137578 )

2024-12-18 15:12:19 +00:00

__init__.py

…

CMakeLists.txt

[aoti] Add initial custom op support (#127034 )

2024-07-24 20:29:55 +00:00

custom_ops.cpp

[AOTI] Fix an issue when fallback op does not return a value (#142339 )

2024-12-09 23:24:29 +00:00

indirect_assert_helper.py

[BE][Easy][12/19] enforce style for empty lines in import segments in test/i*/ (#129763 )

2024-07-18 07:49:19 +00:00

minifier_smoke.py

[BE][Easy][12/19] enforce style for empty lines in import segments in test/i*/ (#129763 )

2024-07-18 07:49:19 +00:00

opinfo_harness.py

[BE][Easy][12/19] enforce style for empty lines in import segments in test/i*/ (#129763 )

2024-07-18 07:49:19 +00:00

s429861_repro.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_aot_inductor_arrayref.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_aot_inductor_custom_ops.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_aot_inductor_package.py

[Codemod][AddExplicitStrictExportArg] caffe2/test/inductor (#143929 )

2024-12-28 23:39:21 +00:00

test_aot_inductor_utils.py

[AOTI XPU] Support AOT Inductor for Intel GPU. (#140269 )

2024-12-10 05:05:08 +00:00

test_aot_inductor.py

Revert "Add AOT inductor support for _scaled_mm for CPU (#141961 )"

2024-12-28 16:49:38 +00:00

test_auto_functionalize.py

Fix another item memo loss location + bool specialization bug (#139587 )

2024-11-09 03:11:19 +00:00

test_autoheuristic.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_b2b_gemm.py

Rename cache limit to recompile limit in configs (#143709 )

2024-12-22 10:03:57 +00:00

test_benchmark_fusion.py

[inductor] Refactor is_big_gpu (#142220 )

2024-12-08 18:51:36 +00:00

test_benchmarking.py

Safely infer device type + docstrings + tests (#133668 )

2024-08-21 05:27:31 +00:00

test_binary_folding.py

[Inductor] add flag for linear binary folding and turn it off by default (#142108 )

2024-12-06 07:12:29 +00:00

test_ck_backend.py

[inductor] Refactor is_big_gpu (#142220 )

2024-12-08 18:51:36 +00:00

test_codecache.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_codegen_triton.py

[Inductor] New Triton Attrs Descriptor Fixups (#138390 )

2024-10-23 14:13:49 +00:00

test_combo_kernels.py

Fix lint errors in fbcode (#135614 )

2024-09-13 02:04:34 +00:00

test_compile_worker.py

[inductor] calibration inductor windows uts (2/N) (#134358 )

2024-08-24 04:08:59 +00:00

test_compiled_autograd.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_compiled_optimizers.py

[foreach_map] Add foreach_map Adam impl to compiled optimizer tests (#143454 )

2024-12-19 03:16:47 +00:00

test_config.py

config: Modify install_config_module to use a layered approach (#138758 )

2024-10-29 23:19:36 +00:00

test_control_flow.py

[hop][BE] unify meta checking with check_meta_consistency (#143545 )

2025-01-03 19:01:07 +00:00

test_cooperative_reductions.py

[Inductor] Fix cooperative reduction tests broken in recent refactor (#143135 )

2024-12-13 02:03:43 +00:00

test_coordinate_descent_tuner.py

[Inductor] Refactor "r" reduction prefix to {"r0_", "r1_"}. (#142020 )

2024-12-12 17:22:20 +00:00

test_cpp_wrapper_hipify.py

Test cpp_wrapper_hipify string comparison (#141353 )

2024-12-03 18:25:32 +00:00

test_cpu_cpp_wrapper.py

[AOTI] add C shim for QConvPointWise (#138540 )

2024-10-31 02:03:01 +00:00

test_cpu_repro.py

[Inductor][lowering] support out_dtype for dequant lowering (#143845 )

2025-01-04 08:48:41 +00:00

test_cpu_select_algorithm.py

[Inductor] add flag for linear binary folding and turn it off by default (#142108 )

2024-12-06 07:12:29 +00:00

test_cuda_repro.py

[inductor] Add missing py312 xfail (#144006 )

2024-12-31 23:37:05 +00:00

test_cudacodecache.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_cudagraph_trees_expandable_segments.py

Revert "Use absolute path path.resolve() -> path.absolute() (#129409 )"

2025-01-04 14:17:20 +00:00

test_cudagraph_trees.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_custom_lowering.py

[Inductor UT] Generalize newly introduced inductor UTs for intel GPU (Part 1) (#136069 )

2024-10-18 16:58:09 +00:00

test_custom_post_grad_passes.py

[inductor] Add API to make post_grad_custom passes cache-able (#137298 )

2024-10-07 21:11:54 +00:00

test_cutlass_backend.py

[AOTI] don't codegen autotune_at_compile_time for non-Triton kernels (#143990 )

2025-01-03 18:01:12 +00:00

test_debug_trace.py

[inductor] Fix 3d tiling (#141709 )

2024-12-01 19:47:41 +00:00

test_decompose_mem_bound_mm.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_dependencies.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_distributed_patterns.py

[ca] expose option to collect sizes as dynamic (#141153 )

2024-11-22 19:26:27 +00:00

test_efficient_conv_bn_eval.py

Revert "Always unspecialize float in OSS (#138922 )"

2024-11-26 00:03:03 +00:00

test_extension_backend.py

Run only listed tests on s390x (#140265 )

2024-11-20 22:53:09 +00:00

test_external_callables.py

[Inductor] External callable registration API for Matmul tuning candidates (#130774 )

2024-10-02 15:38:10 +00:00

test_flex_attention.py

[inductor] Shorten tracebacks for errors inside inductor (by skipping AOTAutograd frames) (#143610 )

2024-12-24 21:48:32 +00:00

test_flex_decoding.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_foreach.py

[foreach_map] Add foreach_map Adam impl to compiled optimizer tests (#143454 )

2024-12-19 03:16:47 +00:00

test_fp8.py

Revert "Add torch._scaled_mm for CPU (#139975 )"

2024-12-28 16:49:38 +00:00

test_fused_attention.py

Fix the Rank of logsumexp Tensor and mGPU support. (#137717 )

2024-10-17 21:58:14 +00:00

test_fx_fusion.py

[inductor] Replace torch.allclose with torch.testing.assert_close in test_fx_fusion (#130618 )

2024-08-06 03:58:43 +00:00

test_gpu_cpp_wrapper.py

[Inductor XPU] Support max-autotune on XPU and reuse the corresponding Inductor UT. (#143266 )

2024-12-30 23:51:17 +00:00

test_graph_transform_observer.py

Fix the Rank of logsumexp Tensor and mGPU support. (#137717 )

2024-10-17 21:58:14 +00:00

test_group_batch_fusion.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_halide.py

Fix lint errors in fbcode (#135614 )

2024-09-13 02:04:34 +00:00

test_indexing.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_inductor_annotations.py

Inductor annotations (#130429 )

2024-12-10 08:53:39 +00:00

test_inductor_freezing.py

[Dynamo] Replace torch._dynamo.optimize() with torch.compile() [10/N] (#142451 )

2024-12-17 12:18:29 +00:00

test_inductor_utils.py

can't infer device on benchmarked function with no args or kwargs (#133290 )

2024-08-13 20:13:44 +00:00

test_inplacing_pass.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_kernel_benchmark.py

[Inductor XPU] Support max-autotune on XPU and reuse the corresponding Inductor UT. (#143266 )

2024-12-30 23:51:17 +00:00

test_layout_optim.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_loop_ordering.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_max_autotune.py

[Inductor XPU] Support max-autotune on XPU and reuse the corresponding Inductor UT. (#143266 )

2024-12-30 23:51:17 +00:00

test_memory_planning.py

Move Sympy printers to torch/utils/_sympy/printers.py (#140597 )

2024-11-26 18:11:00 +00:00

test_memory.py

[inductor] don't fuse two nodes if likely increase peak memory (#138756 )

2024-11-04 20:49:29 +00:00

test_metrics.py

[Inductor] Update AttrsDescriptor instantiation for Triton changes (#137458 )

2024-10-14 20:20:29 +00:00

test_minifier_isolate.py

[Inductor UT] Generalize inductor UT for intel GPU (Part 2) (#134556 )

2024-09-13 05:16:28 +00:00

test_minifier_utils.py

[AOTI Minifier] Save EP instead of graphs (#141159 )

2024-11-22 01:51:10 +00:00

test_minifier.py

Aoti minifier flatten (#141156 )

2024-12-06 07:12:45 +00:00

test_mkldnn_pattern_matcher.py

Enable mkldnn pattern matcher tests for BF16 on AArch64 (#144030 )

2025-01-02 17:13:38 +00:00

test_mmdecomp.py

[BE][Easy][12/19] enforce style for empty lines in import segments in test/i*/ (#129763 )

2024-07-18 07:49:19 +00:00

test_move_constructors_to_cuda.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_mps_basic.py

[MPSInductor] Add masked implementation (#144084 )

2025-01-04 04:30:07 +00:00

test_multi_kernel.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_op_dtype_prop.py

[Inductor/Triton] Upcast FP16/BF16 math reductions to FP32 (#141052 )

2025-01-04 07:57:10 +00:00

test_ordered_set.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_pad_mm.py

Enabled force_shape_pad for test_pad_mm and test_slice_mm_bandwidth_computation (#141768 )

2024-12-24 11:03:39 +00:00

test_padding.py

[Inductor] Refactor "r" reduction prefix to {"r0_", "r1_"}. (#142020 )

2024-12-12 17:22:20 +00:00

test_pattern_matcher.py

Revert "Fix duplicate pattern error (#139321 )"

2024-12-31 17:44:02 +00:00

test_perf.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_profiler.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_provenance_tracing.py

[provenance_tracking] Dump inductor_triton_kernel_to_post_grad_nodes.json info in debug_trace (#143055 )

2024-12-18 06:51:50 +00:00

test_scatter_optimization.py

[BC breaking] move benchmarking + prefer inductor path (#132827 )

2024-08-08 00:47:45 +00:00

test_select_algorithm.py

[Inductor XPU] Support max-autotune on XPU and reuse the corresponding Inductor UT. (#143266 )

2024-12-30 23:51:17 +00:00

test_smoke.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_snode_runtime.py

[Inductor UT] Generalize newly introduced inductor UTs for intel GPU (Part 3) (#136947 )

2024-10-12 13:21:20 +00:00

test_split_cat_fx_passes.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_standalone_compile.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_torchbind.py

Add None return type to init -- tests rest (#132376 )

2024-08-01 15:44:51 +00:00

test_torchinductor_codegen_config_overrides.py

Enable concat support through inductor using pointwise kernels (#141966 )

2024-12-06 14:28:07 +00:00

test_torchinductor_codegen_dynamic_shapes.py

Revert "fix randint distribution for large max (#143787 )"

2024-12-27 09:16:36 +00:00

test_torchinductor_dynamic_shapes.py

Revert "fix randint distribution for large max (#143787 )"

2024-12-27 09:16:36 +00:00

test_torchinductor_opinfo.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_torchinductor_strided_blocks.py

[Inductor] Generalize tiling algorithm to handle fused reductions (#144041 )

2025-01-03 18:16:27 +00:00

test_torchinductor.py

[Inductor UT] Generalize device-bias code in test_torchinductor.py introduced by #143884 . (#144057 )

2025-01-04 02:24:33 +00:00

test_triton_cpu_backend.py

Refactor test_torchinductor_strided_blocks to also support triton CPU (#141587 )

2024-12-05 09:57:08 +00:00

test_triton_extension_backend.py

Have Triton custom extension test use privateuseone device (#137611 )

2024-10-11 21:27:29 +00:00

test_triton_heuristics.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_triton_kernels.py

[user triton] add support for prune_configs_by in @triton.autotune (#142207 )

2025-01-04 03:50:28 +00:00

test_triton_syntax.py

[Inductor UT] Generalize device-bias code in test_triton_syntax.py. (#143178 )

2024-12-14 02:08:32 +00:00

test_triton_wrapper.py

Fix unused Python variables in test/[e-z]* (#136964 )

2024-12-18 23:02:30 +00:00

test_unbacked_symints.py

Enable ruff's unused variable checking everywhere in pytorch (#136965 )

2024-12-22 02:33:11 +00:00

test_utils.py

Proper handling of arguments passed by in kwargs inside zip_schema (#137311 )

2024-10-04 21:50:31 +00:00

test_xpu_basic.py

Fix lint errors in fbcode (#135614 )

2024-09-13 02:04:34 +00:00