├── .github ├── ISSUE_TEMPLATE │ ├── bug_report.md │ └── feature_request.md ├── dependabot.yml ├── matrix.json └── workflows │ ├── benchmark.yml │ ├── benchmark_dispatch.yml │ ├── benchmark_nightly.yml │ ├── compute-benchmark-matrix.yml │ ├── docs-deploy.yml │ ├── docs-test.yml │ ├── lint.yml │ ├── publish-to-pypi.yml │ └── test.yml ├── .gitignore ├── .pre-commit-config.yaml ├── AGENTS.md ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── benchmarks ├── README.md └── run.py ├── docs ├── Makefile ├── README.md ├── _static │ ├── helion_logo.png │ ├── helion_nobackground.png │ └── js │ │ └── runllm-widget.js ├── api │ ├── autotuner.md │ ├── config.md │ ├── exceptions.md │ ├── index.md │ ├── kernel.md │ ├── language.md │ ├── runtime.md │ └── settings.md ├── conf.py ├── deployment_autotuning.md ├── helion_puzzles.rst ├── index.md └── installation.md ├── examples ├── README.rst ├── add.py ├── all_gather_matmul.py ├── all_reduce.py ├── attention.py ├── bf16xint16_gemm.py ├── blackwell_attention.py ├── bmm.py ├── concatenate.py ├── cross_entropy.py ├── embedding.py ├── exp.py ├── fp8_attention.py ├── fp8_gemm.py ├── fused_linear_jsd.py ├── gather_gemv.py ├── gdn_fwd_h.py ├── geglu.py ├── grouped_gemm.py ├── grpo_loss.py ├── int4_gemm.py ├── jagged_dense_add.py ├── jagged_dense_bmm.py ├── jagged_hstu_attn.py ├── jagged_layer_norm.py ├── jagged_mean.py ├── jagged_softmax.py ├── jagged_sum.py ├── jsd.py ├── kl_div.py ├── layer_norm.py ├── long_sum.py ├── low_mem_dropout.py ├── mamba2_chunk_scan.py ├── mamba2_chunk_state.py ├── matmul.py ├── matmul_layernorm.py ├── matmul_split_k.py ├── moe_matmul_ogs.py ├── rms_norm.py ├── segment_reduction.py ├── softmax.py ├── squeeze_and_excitation_net.py ├── sum.py ├── swiglu.py └── welford.py ├── helion ├── __init__.py ├── _compat.py ├── _compiler │ ├── __init__.py │ ├── ast_extension.py │ ├── ast_read_writes.py │ ├── aten_lowering.py │ ├── compile_environment.py │ ├── device_function.py │ ├── device_ir.py │ ├── dtype_utils.py │ ├── generate_ast.py │ ├── helper_function.py │ ├── host_function.py │ ├── indexing_strategy.py │ ├── inductor_lowering.py │ ├── inductor_lowering_extra.py │ ├── lift_closures.py │ ├── loop_dependency_checker.py │ ├── matmul_utils.py │ ├── node_masking.py │ ├── output_header.py │ ├── output_lines.py │ ├── program_id.py │ ├── reduction_strategy.py │ ├── roll_reduction.py │ ├── source_location.py │ ├── static_loop_unroller.py │ ├── tensor_utils.py │ ├── tile_dispatch.py │ ├── tile_strategy.py │ ├── traceback_compat.py │ ├── type_printer.py │ ├── type_propagation.py │ ├── utils.py │ └── variable_origin.py ├── _logging │ ├── __init__.py │ └── _internal.py ├── _testing.py ├── _utils.py ├── autotuner │ ├── __init__.py │ ├── base_cache.py │ ├── base_search.py │ ├── benchmarking.py │ ├── block_id_sequence.py │ ├── config_fragment.py │ ├── config_generation.py │ ├── config_spec.py │ ├── de_surrogate_hybrid.py │ ├── differential_evolution.py │ ├── effort_profile.py │ ├── finite_search.py │ ├── local_cache.py │ ├── logger.py │ ├── pattern_search.py │ ├── progress_bar.py │ ├── random_search.py │ └── surrogate_pattern_search.py ├── exc.py ├── language │ ├── __init__.py │ ├── _decorators.py │ ├── _tracing_ops.py │ ├── atomic_ops.py │ ├── builtin_ops.py │ ├── constexpr.py │ ├── creation_ops.py │ ├── debug_ops.py │ ├── device_print.py │ ├── inline_asm_ops.py │ ├── inline_triton_ops.py │ ├── loops.py │ ├── matmul_ops.py │ ├── memory_ops.py │ ├── random_ops.py │ ├── reduce_ops.py │ ├── ref_tile.py │ ├── scan_ops.py │ ├── signal_wait.py │ ├── stack_tensor.py │ ├── tile_interface.py │ ├── tile_ops.py │ ├── tile_proxy.py │ ├── tunable_ops.py │ └── view_ops.py └── runtime │ ├── __init__.py │ ├── config.py │ ├── kernel.py │ ├── precompile_shim.py │ ├── ref_mode.py │ ├── settings.py │ └── triton_helpers.py ├── lint.sh ├── notebooks └── softmax.ipynb ├── pyproject.toml ├── scripts ├── dictionary.txt ├── lint_examples_main.py └── lint_no_hardcoded_device.py └── test ├── __init__.py ├── data ├── all_ast_nodes.py └── basic_kernels.py ├── test_amd_cdna.py ├── test_associative_scan.expected ├── test_associative_scan.py ├── test_atomic_ops.expected ├── test_atomic_ops.py ├── test_autotuner.expected ├── test_autotuner.py ├── test_breakpoint.py ├── test_broadcasting.expected ├── test_broadcasting.py ├── test_cache.py ├── test_closures.expected ├── test_closures.py ├── test_codegen_comments.py ├── test_config_api.py ├── test_constexpr.expected ├── test_constexpr.py ├── test_control_flow.expected ├── test_control_flow.py ├── test_custom_op.py ├── test_debug_utils.expected ├── test_debug_utils.py ├── test_dot.expected ├── test_dot.py ├── test_dot_requirements.py ├── test_errors.py ├── test_eviction_policy.expected ├── test_eviction_policy.py ├── test_examples.expected ├── test_examples.py ├── test_examples_dist.expected ├── test_examples_dist.py ├── test_generate_ast.expected ├── test_generate_ast.py ├── test_graph_module.expected ├── test_graph_module.py ├── test_grid.expected ├── test_grid.py ├── test_indexing.expected ├── test_indexing.py ├── test_inline_asm_elementwise.expected ├── test_inline_asm_elementwise.py ├── test_inline_triton.expected ├── test_inline_triton.py ├── test_logging.py ├── test_loop_dependencies.py ├── test_loops.expected ├── test_loops.py ├── test_masking.expected ├── test_masking.py ├── test_matmul.expected ├── test_matmul.py ├── test_misc.expected ├── test_misc.py ├── test_persistent_kernels.expected ├── test_persistent_kernels.py ├── test_print.py ├── test_print_ref_eager_mode.py ├── test_random.expected ├── test_random.py ├── test_reduce.expected ├── test_reduce.py ├── test_reductions.expected ├── test_reductions.py ├── test_ref_eager.py ├── test_register_tunable.expected ├── test_register_tunable.py ├── test_rng.expected ├── test_rng.py ├── test_signal_wait.expected ├── test_signal_wait.py ├── test_specialize.expected ├── test_specialize.py ├── test_stack_tensor.expected ├── test_stack_tensor.py ├── test_tensor_descriptor.expected ├── test_tensor_descriptor.py ├── test_torch_compile.py ├── test_triton_kernel.expected ├── test_triton_kernel.py ├── test_type_propagation.expected ├── test_type_propagation.py ├── test_unroll_tuples.expected ├── test_unroll_tuples.py ├── test_views.expected ├── test_views.py └── test_zero_size.py /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/matrix.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/.github/matrix.json -------------------------------------------------------------------------------- /.github/workflows/benchmark.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/.github/workflows/benchmark.yml -------------------------------------------------------------------------------- /.github/workflows/benchmark_dispatch.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/.github/workflows/benchmark_dispatch.yml -------------------------------------------------------------------------------- /.github/workflows/benchmark_nightly.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/.github/workflows/benchmark_nightly.yml -------------------------------------------------------------------------------- /.github/workflows/compute-benchmark-matrix.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/.github/workflows/compute-benchmark-matrix.yml -------------------------------------------------------------------------------- /.github/workflows/docs-deploy.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/.github/workflows/docs-deploy.yml -------------------------------------------------------------------------------- /.github/workflows/docs-test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/.github/workflows/docs-test.yml -------------------------------------------------------------------------------- /.github/workflows/lint.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/.github/workflows/lint.yml -------------------------------------------------------------------------------- /.github/workflows/publish-to-pypi.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/.github/workflows/publish-to-pypi.yml -------------------------------------------------------------------------------- /.github/workflows/test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/.github/workflows/test.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /AGENTS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/AGENTS.md -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/README.md -------------------------------------------------------------------------------- /benchmarks/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/benchmarks/README.md -------------------------------------------------------------------------------- /benchmarks/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/benchmarks/run.py -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/_static/helion_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/docs/_static/helion_logo.png -------------------------------------------------------------------------------- /docs/_static/helion_nobackground.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/docs/_static/helion_nobackground.png -------------------------------------------------------------------------------- /docs/_static/js/runllm-widget.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/docs/_static/js/runllm-widget.js -------------------------------------------------------------------------------- /docs/api/autotuner.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/docs/api/autotuner.md -------------------------------------------------------------------------------- /docs/api/config.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/docs/api/config.md -------------------------------------------------------------------------------- /docs/api/exceptions.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/docs/api/exceptions.md -------------------------------------------------------------------------------- /docs/api/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/docs/api/index.md -------------------------------------------------------------------------------- /docs/api/kernel.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/docs/api/kernel.md -------------------------------------------------------------------------------- /docs/api/language.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/docs/api/language.md -------------------------------------------------------------------------------- /docs/api/runtime.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/docs/api/runtime.md -------------------------------------------------------------------------------- /docs/api/settings.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/docs/api/settings.md -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/deployment_autotuning.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/docs/deployment_autotuning.md -------------------------------------------------------------------------------- /docs/helion_puzzles.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/docs/helion_puzzles.rst -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/docs/index.md -------------------------------------------------------------------------------- /docs/installation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/docs/installation.md -------------------------------------------------------------------------------- /examples/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/README.rst -------------------------------------------------------------------------------- /examples/add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/add.py -------------------------------------------------------------------------------- /examples/all_gather_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/all_gather_matmul.py -------------------------------------------------------------------------------- /examples/all_reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/all_reduce.py -------------------------------------------------------------------------------- /examples/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/attention.py -------------------------------------------------------------------------------- /examples/bf16xint16_gemm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/bf16xint16_gemm.py -------------------------------------------------------------------------------- /examples/blackwell_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/blackwell_attention.py -------------------------------------------------------------------------------- /examples/bmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/bmm.py -------------------------------------------------------------------------------- /examples/concatenate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/concatenate.py -------------------------------------------------------------------------------- /examples/cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/cross_entropy.py -------------------------------------------------------------------------------- /examples/embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/embedding.py -------------------------------------------------------------------------------- /examples/exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/exp.py -------------------------------------------------------------------------------- /examples/fp8_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/fp8_attention.py -------------------------------------------------------------------------------- /examples/fp8_gemm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/fp8_gemm.py -------------------------------------------------------------------------------- /examples/fused_linear_jsd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/fused_linear_jsd.py -------------------------------------------------------------------------------- /examples/gather_gemv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/gather_gemv.py -------------------------------------------------------------------------------- /examples/gdn_fwd_h.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/gdn_fwd_h.py -------------------------------------------------------------------------------- /examples/geglu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/geglu.py -------------------------------------------------------------------------------- /examples/grouped_gemm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/grouped_gemm.py -------------------------------------------------------------------------------- /examples/grpo_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/grpo_loss.py -------------------------------------------------------------------------------- /examples/int4_gemm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/int4_gemm.py -------------------------------------------------------------------------------- /examples/jagged_dense_add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/jagged_dense_add.py -------------------------------------------------------------------------------- /examples/jagged_dense_bmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/jagged_dense_bmm.py -------------------------------------------------------------------------------- /examples/jagged_hstu_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/jagged_hstu_attn.py -------------------------------------------------------------------------------- /examples/jagged_layer_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/jagged_layer_norm.py -------------------------------------------------------------------------------- /examples/jagged_mean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/jagged_mean.py -------------------------------------------------------------------------------- /examples/jagged_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/jagged_softmax.py -------------------------------------------------------------------------------- /examples/jagged_sum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/jagged_sum.py -------------------------------------------------------------------------------- /examples/jsd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/jsd.py -------------------------------------------------------------------------------- /examples/kl_div.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/kl_div.py -------------------------------------------------------------------------------- /examples/layer_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/layer_norm.py -------------------------------------------------------------------------------- /examples/long_sum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/long_sum.py -------------------------------------------------------------------------------- /examples/low_mem_dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/low_mem_dropout.py -------------------------------------------------------------------------------- /examples/mamba2_chunk_scan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/mamba2_chunk_scan.py -------------------------------------------------------------------------------- /examples/mamba2_chunk_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/mamba2_chunk_state.py -------------------------------------------------------------------------------- /examples/matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/matmul.py -------------------------------------------------------------------------------- /examples/matmul_layernorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/matmul_layernorm.py -------------------------------------------------------------------------------- /examples/matmul_split_k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/matmul_split_k.py -------------------------------------------------------------------------------- /examples/moe_matmul_ogs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/moe_matmul_ogs.py -------------------------------------------------------------------------------- /examples/rms_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/rms_norm.py -------------------------------------------------------------------------------- /examples/segment_reduction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/segment_reduction.py -------------------------------------------------------------------------------- /examples/softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/softmax.py -------------------------------------------------------------------------------- /examples/squeeze_and_excitation_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/squeeze_and_excitation_net.py -------------------------------------------------------------------------------- /examples/sum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/sum.py -------------------------------------------------------------------------------- /examples/swiglu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/swiglu.py -------------------------------------------------------------------------------- /examples/welford.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/examples/welford.py -------------------------------------------------------------------------------- /helion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/__init__.py -------------------------------------------------------------------------------- /helion/_compat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compat.py -------------------------------------------------------------------------------- /helion/_compiler/__init__.py: -------------------------------------------------------------------------------- 1 | from __future__ import annotations 2 | -------------------------------------------------------------------------------- /helion/_compiler/ast_extension.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/ast_extension.py -------------------------------------------------------------------------------- /helion/_compiler/ast_read_writes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/ast_read_writes.py -------------------------------------------------------------------------------- /helion/_compiler/aten_lowering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/aten_lowering.py -------------------------------------------------------------------------------- /helion/_compiler/compile_environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/compile_environment.py -------------------------------------------------------------------------------- /helion/_compiler/device_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/device_function.py -------------------------------------------------------------------------------- /helion/_compiler/device_ir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/device_ir.py -------------------------------------------------------------------------------- /helion/_compiler/dtype_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/dtype_utils.py -------------------------------------------------------------------------------- /helion/_compiler/generate_ast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/generate_ast.py -------------------------------------------------------------------------------- /helion/_compiler/helper_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/helper_function.py -------------------------------------------------------------------------------- /helion/_compiler/host_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/host_function.py -------------------------------------------------------------------------------- /helion/_compiler/indexing_strategy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/indexing_strategy.py -------------------------------------------------------------------------------- /helion/_compiler/inductor_lowering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/inductor_lowering.py -------------------------------------------------------------------------------- /helion/_compiler/inductor_lowering_extra.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/inductor_lowering_extra.py -------------------------------------------------------------------------------- /helion/_compiler/lift_closures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/lift_closures.py -------------------------------------------------------------------------------- /helion/_compiler/loop_dependency_checker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/loop_dependency_checker.py -------------------------------------------------------------------------------- /helion/_compiler/matmul_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/matmul_utils.py -------------------------------------------------------------------------------- /helion/_compiler/node_masking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/node_masking.py -------------------------------------------------------------------------------- /helion/_compiler/output_header.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/output_header.py -------------------------------------------------------------------------------- /helion/_compiler/output_lines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/output_lines.py -------------------------------------------------------------------------------- /helion/_compiler/program_id.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/program_id.py -------------------------------------------------------------------------------- /helion/_compiler/reduction_strategy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/reduction_strategy.py -------------------------------------------------------------------------------- /helion/_compiler/roll_reduction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/roll_reduction.py -------------------------------------------------------------------------------- /helion/_compiler/source_location.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/source_location.py -------------------------------------------------------------------------------- /helion/_compiler/static_loop_unroller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/static_loop_unroller.py -------------------------------------------------------------------------------- /helion/_compiler/tensor_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/tensor_utils.py -------------------------------------------------------------------------------- /helion/_compiler/tile_dispatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/tile_dispatch.py -------------------------------------------------------------------------------- /helion/_compiler/tile_strategy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/tile_strategy.py -------------------------------------------------------------------------------- /helion/_compiler/traceback_compat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/traceback_compat.py -------------------------------------------------------------------------------- /helion/_compiler/type_printer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/type_printer.py -------------------------------------------------------------------------------- /helion/_compiler/type_propagation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/type_propagation.py -------------------------------------------------------------------------------- /helion/_compiler/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/utils.py -------------------------------------------------------------------------------- /helion/_compiler/variable_origin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_compiler/variable_origin.py -------------------------------------------------------------------------------- /helion/_logging/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_logging/__init__.py -------------------------------------------------------------------------------- /helion/_logging/_internal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_logging/_internal.py -------------------------------------------------------------------------------- /helion/_testing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_testing.py -------------------------------------------------------------------------------- /helion/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/_utils.py -------------------------------------------------------------------------------- /helion/autotuner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/autotuner/__init__.py -------------------------------------------------------------------------------- /helion/autotuner/base_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/autotuner/base_cache.py -------------------------------------------------------------------------------- /helion/autotuner/base_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/autotuner/base_search.py -------------------------------------------------------------------------------- /helion/autotuner/benchmarking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/autotuner/benchmarking.py -------------------------------------------------------------------------------- /helion/autotuner/block_id_sequence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/autotuner/block_id_sequence.py -------------------------------------------------------------------------------- /helion/autotuner/config_fragment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/autotuner/config_fragment.py -------------------------------------------------------------------------------- /helion/autotuner/config_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/autotuner/config_generation.py -------------------------------------------------------------------------------- /helion/autotuner/config_spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/autotuner/config_spec.py -------------------------------------------------------------------------------- /helion/autotuner/de_surrogate_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/autotuner/de_surrogate_hybrid.py -------------------------------------------------------------------------------- /helion/autotuner/differential_evolution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/autotuner/differential_evolution.py -------------------------------------------------------------------------------- /helion/autotuner/effort_profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/autotuner/effort_profile.py -------------------------------------------------------------------------------- /helion/autotuner/finite_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/autotuner/finite_search.py -------------------------------------------------------------------------------- /helion/autotuner/local_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/autotuner/local_cache.py -------------------------------------------------------------------------------- /helion/autotuner/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/autotuner/logger.py -------------------------------------------------------------------------------- /helion/autotuner/pattern_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/autotuner/pattern_search.py -------------------------------------------------------------------------------- /helion/autotuner/progress_bar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/autotuner/progress_bar.py -------------------------------------------------------------------------------- /helion/autotuner/random_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/autotuner/random_search.py -------------------------------------------------------------------------------- /helion/autotuner/surrogate_pattern_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/autotuner/surrogate_pattern_search.py -------------------------------------------------------------------------------- /helion/exc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/exc.py -------------------------------------------------------------------------------- /helion/language/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/__init__.py -------------------------------------------------------------------------------- /helion/language/_decorators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/_decorators.py -------------------------------------------------------------------------------- /helion/language/_tracing_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/_tracing_ops.py -------------------------------------------------------------------------------- /helion/language/atomic_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/atomic_ops.py -------------------------------------------------------------------------------- /helion/language/builtin_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/builtin_ops.py -------------------------------------------------------------------------------- /helion/language/constexpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/constexpr.py -------------------------------------------------------------------------------- /helion/language/creation_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/creation_ops.py -------------------------------------------------------------------------------- /helion/language/debug_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/debug_ops.py -------------------------------------------------------------------------------- /helion/language/device_print.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/device_print.py -------------------------------------------------------------------------------- /helion/language/inline_asm_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/inline_asm_ops.py -------------------------------------------------------------------------------- /helion/language/inline_triton_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/inline_triton_ops.py -------------------------------------------------------------------------------- /helion/language/loops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/loops.py -------------------------------------------------------------------------------- /helion/language/matmul_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/matmul_ops.py -------------------------------------------------------------------------------- /helion/language/memory_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/memory_ops.py -------------------------------------------------------------------------------- /helion/language/random_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/random_ops.py -------------------------------------------------------------------------------- /helion/language/reduce_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/reduce_ops.py -------------------------------------------------------------------------------- /helion/language/ref_tile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/ref_tile.py -------------------------------------------------------------------------------- /helion/language/scan_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/scan_ops.py -------------------------------------------------------------------------------- /helion/language/signal_wait.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/signal_wait.py -------------------------------------------------------------------------------- /helion/language/stack_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/stack_tensor.py -------------------------------------------------------------------------------- /helion/language/tile_interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/tile_interface.py -------------------------------------------------------------------------------- /helion/language/tile_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/tile_ops.py -------------------------------------------------------------------------------- /helion/language/tile_proxy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/tile_proxy.py -------------------------------------------------------------------------------- /helion/language/tunable_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/tunable_ops.py -------------------------------------------------------------------------------- /helion/language/view_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/language/view_ops.py -------------------------------------------------------------------------------- /helion/runtime/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/runtime/__init__.py -------------------------------------------------------------------------------- /helion/runtime/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/runtime/config.py -------------------------------------------------------------------------------- /helion/runtime/kernel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/runtime/kernel.py -------------------------------------------------------------------------------- /helion/runtime/precompile_shim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/runtime/precompile_shim.py -------------------------------------------------------------------------------- /helion/runtime/ref_mode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/runtime/ref_mode.py -------------------------------------------------------------------------------- /helion/runtime/settings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/runtime/settings.py -------------------------------------------------------------------------------- /helion/runtime/triton_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/helion/runtime/triton_helpers.py -------------------------------------------------------------------------------- /lint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/lint.sh -------------------------------------------------------------------------------- /notebooks/softmax.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/notebooks/softmax.ipynb -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/pyproject.toml -------------------------------------------------------------------------------- /scripts/dictionary.txt: -------------------------------------------------------------------------------- 1 | NotIn 2 | readd 3 | -------------------------------------------------------------------------------- /scripts/lint_examples_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/scripts/lint_examples_main.py -------------------------------------------------------------------------------- /scripts/lint_no_hardcoded_device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/scripts/lint_no_hardcoded_device.py -------------------------------------------------------------------------------- /test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/data/all_ast_nodes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/data/all_ast_nodes.py -------------------------------------------------------------------------------- /test/data/basic_kernels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/data/basic_kernels.py -------------------------------------------------------------------------------- /test/test_amd_cdna.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_amd_cdna.py -------------------------------------------------------------------------------- /test/test_associative_scan.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_associative_scan.expected -------------------------------------------------------------------------------- /test/test_associative_scan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_associative_scan.py -------------------------------------------------------------------------------- /test/test_atomic_ops.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_atomic_ops.expected -------------------------------------------------------------------------------- /test/test_atomic_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_atomic_ops.py -------------------------------------------------------------------------------- /test/test_autotuner.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_autotuner.expected -------------------------------------------------------------------------------- /test/test_autotuner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_autotuner.py -------------------------------------------------------------------------------- /test/test_breakpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_breakpoint.py -------------------------------------------------------------------------------- /test/test_broadcasting.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_broadcasting.expected -------------------------------------------------------------------------------- /test/test_broadcasting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_broadcasting.py -------------------------------------------------------------------------------- /test/test_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_cache.py -------------------------------------------------------------------------------- /test/test_closures.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_closures.expected -------------------------------------------------------------------------------- /test/test_closures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_closures.py -------------------------------------------------------------------------------- /test/test_codegen_comments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_codegen_comments.py -------------------------------------------------------------------------------- /test/test_config_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_config_api.py -------------------------------------------------------------------------------- /test/test_constexpr.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_constexpr.expected -------------------------------------------------------------------------------- /test/test_constexpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_constexpr.py -------------------------------------------------------------------------------- /test/test_control_flow.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_control_flow.expected -------------------------------------------------------------------------------- /test/test_control_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_control_flow.py -------------------------------------------------------------------------------- /test/test_custom_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_custom_op.py -------------------------------------------------------------------------------- /test/test_debug_utils.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_debug_utils.expected -------------------------------------------------------------------------------- /test/test_debug_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_debug_utils.py -------------------------------------------------------------------------------- /test/test_dot.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_dot.expected -------------------------------------------------------------------------------- /test/test_dot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_dot.py -------------------------------------------------------------------------------- /test/test_dot_requirements.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_dot_requirements.py -------------------------------------------------------------------------------- /test/test_errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_errors.py -------------------------------------------------------------------------------- /test/test_eviction_policy.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_eviction_policy.expected -------------------------------------------------------------------------------- /test/test_eviction_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_eviction_policy.py -------------------------------------------------------------------------------- /test/test_examples.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_examples.expected -------------------------------------------------------------------------------- /test/test_examples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_examples.py -------------------------------------------------------------------------------- /test/test_examples_dist.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_examples_dist.expected -------------------------------------------------------------------------------- /test/test_examples_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_examples_dist.py -------------------------------------------------------------------------------- /test/test_generate_ast.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_generate_ast.expected -------------------------------------------------------------------------------- /test/test_generate_ast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_generate_ast.py -------------------------------------------------------------------------------- /test/test_graph_module.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_graph_module.expected -------------------------------------------------------------------------------- /test/test_graph_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_graph_module.py -------------------------------------------------------------------------------- /test/test_grid.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_grid.expected -------------------------------------------------------------------------------- /test/test_grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_grid.py -------------------------------------------------------------------------------- /test/test_indexing.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_indexing.expected -------------------------------------------------------------------------------- /test/test_indexing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_indexing.py -------------------------------------------------------------------------------- /test/test_inline_asm_elementwise.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_inline_asm_elementwise.expected -------------------------------------------------------------------------------- /test/test_inline_asm_elementwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_inline_asm_elementwise.py -------------------------------------------------------------------------------- /test/test_inline_triton.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_inline_triton.expected -------------------------------------------------------------------------------- /test/test_inline_triton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_inline_triton.py -------------------------------------------------------------------------------- /test/test_logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_logging.py -------------------------------------------------------------------------------- /test/test_loop_dependencies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_loop_dependencies.py -------------------------------------------------------------------------------- /test/test_loops.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_loops.expected -------------------------------------------------------------------------------- /test/test_loops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_loops.py -------------------------------------------------------------------------------- /test/test_masking.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_masking.expected -------------------------------------------------------------------------------- /test/test_masking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_masking.py -------------------------------------------------------------------------------- /test/test_matmul.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_matmul.expected -------------------------------------------------------------------------------- /test/test_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_matmul.py -------------------------------------------------------------------------------- /test/test_misc.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_misc.expected -------------------------------------------------------------------------------- /test/test_misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_misc.py -------------------------------------------------------------------------------- /test/test_persistent_kernels.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_persistent_kernels.expected -------------------------------------------------------------------------------- /test/test_persistent_kernels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_persistent_kernels.py -------------------------------------------------------------------------------- /test/test_print.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_print.py -------------------------------------------------------------------------------- /test/test_print_ref_eager_mode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_print_ref_eager_mode.py -------------------------------------------------------------------------------- /test/test_random.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_random.expected -------------------------------------------------------------------------------- /test/test_random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_random.py -------------------------------------------------------------------------------- /test/test_reduce.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_reduce.expected -------------------------------------------------------------------------------- /test/test_reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_reduce.py -------------------------------------------------------------------------------- /test/test_reductions.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_reductions.expected -------------------------------------------------------------------------------- /test/test_reductions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_reductions.py -------------------------------------------------------------------------------- /test/test_ref_eager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_ref_eager.py -------------------------------------------------------------------------------- /test/test_register_tunable.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_register_tunable.expected -------------------------------------------------------------------------------- /test/test_register_tunable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_register_tunable.py -------------------------------------------------------------------------------- /test/test_rng.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_rng.expected -------------------------------------------------------------------------------- /test/test_rng.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_rng.py -------------------------------------------------------------------------------- /test/test_signal_wait.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_signal_wait.expected -------------------------------------------------------------------------------- /test/test_signal_wait.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_signal_wait.py -------------------------------------------------------------------------------- /test/test_specialize.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_specialize.expected -------------------------------------------------------------------------------- /test/test_specialize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_specialize.py -------------------------------------------------------------------------------- /test/test_stack_tensor.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_stack_tensor.expected -------------------------------------------------------------------------------- /test/test_stack_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_stack_tensor.py -------------------------------------------------------------------------------- /test/test_tensor_descriptor.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_tensor_descriptor.expected -------------------------------------------------------------------------------- /test/test_tensor_descriptor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_tensor_descriptor.py -------------------------------------------------------------------------------- /test/test_torch_compile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_torch_compile.py -------------------------------------------------------------------------------- /test/test_triton_kernel.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_triton_kernel.expected -------------------------------------------------------------------------------- /test/test_triton_kernel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_triton_kernel.py -------------------------------------------------------------------------------- /test/test_type_propagation.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_type_propagation.expected -------------------------------------------------------------------------------- /test/test_type_propagation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_type_propagation.py -------------------------------------------------------------------------------- /test/test_unroll_tuples.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_unroll_tuples.expected -------------------------------------------------------------------------------- /test/test_unroll_tuples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_unroll_tuples.py -------------------------------------------------------------------------------- /test/test_views.expected: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_views.expected -------------------------------------------------------------------------------- /test/test_views.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_views.py -------------------------------------------------------------------------------- /test/test_zero_size.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/helion/HEAD/test/test_zero_size.py --------------------------------------------------------------------------------