| .. |
|
cpp
|
.
|
2025-09-20 16:09:34 +08:00 |
|
extension_backends
|
.
|
2025-09-20 16:09:34 +08:00 |
|
CMakeLists.txt
|
.
|
2025-09-20 16:09:34 +08:00 |
|
__init__.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
custom_ops.cpp
|
.
|
2025-09-20 16:09:34 +08:00 |
|
indirect_assert_helper.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
minifier_smoke.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
opinfo_harness.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
s429861_repro.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_alignment.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_aot_inductor.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_aot_inductor_arrayref.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_aot_inductor_custom_ops.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_aot_inductor_package.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_aot_inductor_utils.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_async_compile.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_auto_functionalize.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_autoheuristic.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_b2b_gemm.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_benchmark_fusion.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_benchmarking.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_best_config.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_binary_folding.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_block_analysis.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_ck_backend.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_codecache.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_codegen_triton.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_combo_kernels.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_compile.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_compile_subprocess.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_compile_worker.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_compiled_autograd.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_compiled_optimizers.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_config.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_control_flow.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_cooperative_reductions.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_coordinate_descent_tuner.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_cpp_wrapper_hipify.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_cpu_cpp_wrapper.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_cpu_repro.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_cpu_select_algorithm.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_cuda_repro.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_cudacodecache.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_cudagraph_trees.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_cudagraph_trees_expandable_segments.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_custom_lowering.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_custom_post_grad_passes.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_cutlass_backend.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_cutlass_evt.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_debug_trace.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_decompose_mem_bound_mm.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_dependencies.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_distributed_patterns.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_efficient_conv_bn_eval.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_extension_backend.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_external_callables.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_flex_attention.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_flex_decoding.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_foreach.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_fp8.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_fused_attention.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_fuzzer.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_fx_fusion.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_fxir_backend.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_gpu_cpp_wrapper.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_graph_transform_observer.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_group_batch_fusion.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_halide.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_helion_kernels.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_indexing.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_inductor_annotations.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_inductor_freezing.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_inductor_scheduler.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_inductor_utils.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_inplace_padding.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_inplacing_pass.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_kernel_benchmark.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_layout_optim.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_loop_ordering.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_max_autotune.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_memory.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_memory_planning.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_metrics.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_minifier.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_minifier_isolate.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_minifier_utils.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_mkldnn_pattern_matcher.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_mmdecomp.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_move_constructors_to_cuda.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_mps_basic.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_multi_kernel.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_online_softmax.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_op_completeness.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_op_dtype_prop.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_ordered_set.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_pad_mm.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_padding.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_pattern_matcher.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_perf.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_profiler.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_provenance_tracing.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_quantization.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_scatter_optimization.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_select_algorithm.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_smoke.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_snode_runtime.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_split_cat_fx_aten_passes.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_split_cat_fx_passes.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_static_cuda_launcher.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_subgraph_choice.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_torchbind.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_torchinductor.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_torchinductor_codegen_config_overrides.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_torchinductor_codegen_dynamic_shapes.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_torchinductor_dynamic_shapes.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_torchinductor_opinfo.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_torchinductor_strided_blocks.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_triton_cpu_backend.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_triton_extension_backend.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_triton_heuristics.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_triton_kernels.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_triton_syntax.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_triton_wrapper.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_unbacked_symints.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_utils.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
test_xpu_basic.py
|
.
|
2025-09-20 16:09:34 +08:00 |