.clang-format
.gitignore
.gitmodules
.pre-commit-config.yaml
Dockerfile
LICENSE
MANIFEST.in
README.md
pyproject.toml
.github/workflows/ci.yml
.github/workflows/release.yml
docs/SPEC-RULES.md
docs/develop.md
docs/plans/TEMPLATE.md
docs/policies/policy.schema.json
docs/policies/project-policy.json
docs/spec/analysis.md
docs/spec/architecture.md
docs/spec/cli.md
docs/spec/code-organization.md
docs/spec/codegen.md
docs/spec/core-ir.md
docs/spec/evaluator.md
docs/spec/hir.md
docs/spec/inspection.md
docs/spec/parser.md
docs/spec/passes.md
docs/spec/runtime.md
docs/spec/schedule.md
docs/spec/semantic-analysis.md
docs/spec/shard.md
docs/spec/target.md
docs/spec/tir.md
docs/spec/types.md
docs/spec/visitor-mutator.md
docs/spec/visitor-registry.md
docs/tutorial/index.md
docs/tutorial/migrate.md
docs/tutorial/optimize.md
include/tilefoundry/runtime.h
include/tilefoundry/runtime/cpu/runtime.h
include/tilefoundry/runtime/cuda/runtime.cuh
include/tilefoundry/runtime/cuda/layout/shard_layout.cuh
include/tilefoundry/runtime/cuda/ops/binary.cuh
include/tilefoundry/runtime/cuda/ops/cast.cuh
include/tilefoundry/runtime/cuda/ops/clamp.cuh
include/tilefoundry/runtime/cuda/ops/copy.cuh
include/tilefoundry/runtime/cuda/ops/fill.cuh
include/tilefoundry/runtime/cuda/ops/mma.cuh
include/tilefoundry/runtime/cuda/ops/reduce.cuh
include/tilefoundry/runtime/cuda/ops/rmsnorm.cuh
include/tilefoundry/runtime/cuda/ops/sync.cuh
include/tilefoundry/runtime/cuda/ops/unary.cuh
include/tilefoundry/runtime/cuda/ops/binary/binary_impl.h
include/tilefoundry/runtime/cuda/ops/copy/copy_impl.h
include/tilefoundry/runtime/cuda/ops/fill/fill_impl.h
include/tilefoundry/runtime/cuda/ops/mma/mma_impl.h
include/tilefoundry/runtime/cuda/ops/reduce/reduce_common_impl.h
include/tilefoundry/runtime/cuda/ops/reduce/reduce_cross_warp_impl.h
include/tilefoundry/runtime/cuda/ops/reduce/reduce_intra_cta_impl.h
include/tilefoundry/runtime/cuda/ops/reduce/reduce_intra_warp_impl.h
include/tilefoundry/runtime/cuda/ops/reduce/reduce_plain_impl.h
include/tilefoundry/runtime/cuda/ops/rmsnorm/rmsnorm_impl.h
include/tilefoundry/runtime/cuda/ops/sync/sync_impl.h
include/tilefoundry/runtime/cuda/ops/unary/unary_impl.h
include/tilefoundry/runtime/cuda/tensor_view/ops_detail.cuh
include/tilefoundry/runtime/cuda/tensor_view/shard_copy.cuh
include/tilefoundry/runtime/cuda/tensor_view/shard_tensor.cuh
scripts/comment_hygiene_lint.py
scripts/english_only_lint.py
scripts/fetch_viewer_assets.py
scripts/finalize_plan_context.py
scripts/forward_references_lint.py
scripts/generate_model_catalog.py
scripts/get_policy.py
scripts/no_machine_paths_lint.py
scripts/spec_entropy_lint.py
scripts/spec_refs_lint.py
scripts/spec_rules_lint.py
scripts/summarise_model_coverage.py
src/tilefoundry/__init__.py
src/tilefoundry/compile.py
src/tilefoundry/module.py
src/tilefoundry/registry.py
src/tilefoundry/script.py
src/tilefoundry.egg-info/PKG-INFO
src/tilefoundry.egg-info/SOURCES.txt
src/tilefoundry.egg-info/dependency_links.txt
src/tilefoundry.egg-info/entry_points.txt
src/tilefoundry.egg-info/requires.txt
src/tilefoundry.egg-info/scm_file_list.json
src/tilefoundry.egg-info/scm_version.json
src/tilefoundry.egg-info/top_level.txt
src/tilefoundry/analysis/__init__.py
src/tilefoundry/analysis/api.py
src/tilefoundry/analysis/compute_cost.py
src/tilefoundry/analysis/errors.py
src/tilefoundry/analysis/facts.py
src/tilefoundry/analysis/memory.py
src/tilefoundry/analysis/metadata.py
src/tilefoundry/analysis/poly.py
src/tilefoundry/analysis/preflight.py
src/tilefoundry/analysis/registry.py
src/tilefoundry/analysis/roofline.py
src/tilefoundry/analysis/timeline.py
src/tilefoundry/analysis/walk.py
src/tilefoundry/cli/__init__.py
src/tilefoundry/cli/__main__.py
src/tilefoundry/cli/analyze.py
src/tilefoundry/cli/check.py
src/tilefoundry/cli/data.py
src/tilefoundry/cli/inspect.py
src/tilefoundry/cli/models.py
src/tilefoundry/cli/schedule.py
src/tilefoundry/cli/source.py
src/tilefoundry/cli/spec.py
src/tilefoundry/cli/tutorial.py
src/tilefoundry/codegen/__init__.py
src/tilefoundry/codegen/linkable.py
src/tilefoundry/codegen/linker.py
src/tilefoundry/codegen/registry.py
src/tilefoundry/codegen/cpu/__init__.py
src/tilefoundry/codegen/cpu/module.py
src/tilefoundry/codegen/cpu/templates/__init__.py
src/tilefoundry/codegen/cpu/templates/cpu_module.cpp.j2
src/tilefoundry/codegen/cuda/__init__.py
src/tilefoundry/codegen/cuda/context.py
src/tilefoundry/codegen/cuda/emit.py
src/tilefoundry/codegen/cuda/module.py
src/tilefoundry/codegen/cuda/templates/__init__.py
src/tilefoundry/codegen/cuda/templates/cuda_module.cu.j2
src/tilefoundry/codegen/cuda/templates/device_kernel.cu.j2
src/tilefoundry/codegen/cuda/tir/__init__.py
src/tilefoundry/codegen/cuda/tir/arith.py
src/tilefoundry/codegen/cuda/tir/async_copy.py
src/tilefoundry/codegen/cuda/tir/clamp.py
src/tilefoundry/codegen/cuda/tir/prim_function.py
src/tilefoundry/codegen/cuda/tir/reduce.py
src/tilefoundry/codegen/cuda/tir/sync.py
src/tilefoundry/codegen/cuda/tir/memory/__init__.py
src/tilefoundry/codegen/cuda/tir/memory/alloc_tensor.py
src/tilefoundry/codegen/cuda/tir/memory/copy.py
src/tilefoundry/codegen/cuda/tir/memory/fill.py
src/tilefoundry/codegen/cuda/tir/memory/ptr_of.py
src/tilefoundry/codegen/cuda/tir/memory/tensor_view.py
src/tilefoundry/codegen/cuda/tir/nn/__init__.py
src/tilefoundry/codegen/cuda/tir/nn/mma.py
src/tilefoundry/codegen/cuda/tir/nn/relu.py
src/tilefoundry/codegen/cuda/tir/nn/rms_norm.py
src/tilefoundry/codegen/cuda/tir/stmts/__init__.py
src/tilefoundry/codegen/cuda/tir/stmts/dispatch_call.py
src/tilefoundry/codegen/cuda/tir/stmts/for_.py
src/tilefoundry/codegen/cuda/tir/stmts/if_.py
src/tilefoundry/codegen/cuda/tir/stmts/let_stmt.py
src/tilefoundry/codegen/cuda/tir/stmts/mesh_scope.py
src/tilefoundry/codegen/cuda/tir/stmts/return_.py
src/tilefoundry/codegen/cuda/tir/stmts/sequential.py
src/tilefoundry/codegen/templates/cmake/CMakeLists.txt.j2
src/tilefoundry/dsl/__init__.py
src/tilefoundry/dsl/__main__.py
src/tilefoundry/dsl/_namespace.py
src/tilefoundry/dsl/_stub_gen.py
src/tilefoundry/dsl/_tensor.py
src/tilefoundry/dsl/py.typed
src/tilefoundry/dsl/storage.py
src/tilefoundry/dsl/T/__init__.py
src/tilefoundry/dsl/T/_platforms.py
src/tilefoundry/dsl/tf/__init__.py
src/tilefoundry/dump/__init__.py
src/tilefoundry/dump/dumpers.py
src/tilefoundry/dump/flags.py
src/tilefoundry/dump/scope.py
src/tilefoundry/evaluator/__init__.py
src/tilefoundry/evaluator/context.py
src/tilefoundry/evaluator/dim.py
src/tilefoundry/evaluator/interpreter.py
src/tilefoundry/evaluator/registry.py
src/tilefoundry/evaluator/value.py
src/tilefoundry/inspection/__init__.py
src/tilefoundry/inspection/analysis_report.py
src/tilefoundry/inspection/dot.py
src/tilefoundry/inspection/python_printer.py
src/tilefoundry/inspection/tir_print.py
src/tilefoundry/inspection/viewer/__init__.py
src/tilefoundry/inspection/viewer/assets.py
src/tilefoundry/inspection/viewer/builder.py
src/tilefoundry/inspection/viewer/htmltable.py
src/tilefoundry/inspection/viewer/palette.py
src/tilefoundry/inspection/viewer/server.py
src/tilefoundry/inspection/viewer/static/.gitignore
src/tilefoundry/inspection/viewer/static/index.html
src/tilefoundry/inspection/viewer/static/viewer.js
src/tilefoundry/ir/__init__.py
src/tilefoundry/ir/visitor.py
src/tilefoundry/ir/constraints/__init__.py
src/tilefoundry/ir/constraints/base.py
src/tilefoundry/ir/constraints/layout.py
src/tilefoundry/ir/constraints/mesh.py
src/tilefoundry/ir/constraints/storage.py
src/tilefoundry/ir/core/__init__.py
src/tilefoundry/ir/core/context.py
src/tilefoundry/ir/core/errors.py
src/tilefoundry/ir/core/expr.py
src/tilefoundry/ir/core/kinds.py
src/tilefoundry/ir/core/metadata.py
src/tilefoundry/ir/core/module.py
src/tilefoundry/ir/core/op.py
src/tilefoundry/ir/core/op_registry.py
src/tilefoundry/ir/core/op_schema.py
src/tilefoundry/ir/core/param_def.py
src/tilefoundry/ir/core/pattern.py
src/tilefoundry/ir/core/register.py
src/tilefoundry/ir/core/stmt.py
src/tilefoundry/ir/hir/__init__.py
src/tilefoundry/ir/hir/_helpers.py
src/tilefoundry/ir/hir/_shard_checks.py
src/tilefoundry/ir/hir/function.py
src/tilefoundry/ir/hir/grid_region.py
src/tilefoundry/ir/hir/specialize.py
src/tilefoundry/ir/hir/verify.py
src/tilefoundry/ir/hir/cuda/__init__.py
src/tilefoundry/ir/hir/cuda/nn/__init__.py
src/tilefoundry/ir/hir/cuda/nn/mma.py
src/tilefoundry/ir/hir/math/__init__.py
src/tilefoundry/ir/hir/math/aliases.py
src/tilefoundry/ir/hir/math/binary.py
src/tilefoundry/ir/hir/math/clamp.py
src/tilefoundry/ir/hir/math/softplus.py
src/tilefoundry/ir/hir/math/unary.py
src/tilefoundry/ir/hir/nn/__init__.py
src/tilefoundry/ir/hir/nn/conv2d.py
src/tilefoundry/ir/hir/nn/gelu.py
src/tilefoundry/ir/hir/nn/layer_norm.py
src/tilefoundry/ir/hir/nn/matmul.py
src/tilefoundry/ir/hir/nn/relu.py
src/tilefoundry/ir/hir/nn/rms_norm.py
src/tilefoundry/ir/hir/nn/rope.py
src/tilefoundry/ir/hir/nn/sigmoid.py
src/tilefoundry/ir/hir/nn/silu.py
src/tilefoundry/ir/hir/nn/softmax.py
src/tilefoundry/ir/hir/nn/tanh.py
src/tilefoundry/ir/hir/shape/__init__.py
src/tilefoundry/ir/hir/shape/shape_compose.py
src/tilefoundry/ir/hir/shape/shape_extract.py
src/tilefoundry/ir/hir/sharding/__init__.py
src/tilefoundry/ir/hir/sharding/local.py
src/tilefoundry/ir/hir/sharding/reshard.py
src/tilefoundry/ir/hir/tensor/__init__.py
src/tilefoundry/ir/hir/tensor/argmax.py
src/tilefoundry/ir/hir/tensor/cache_update.py
src/tilefoundry/ir/hir/tensor/cast.py
src/tilefoundry/ir/hir/tensor/concat.py
src/tilefoundry/ir/hir/tensor/full_like.py
src/tilefoundry/ir/hir/tensor/gather.py
src/tilefoundry/ir/hir/tensor/insert_slice.py
src/tilefoundry/ir/hir/tensor/quant.py
src/tilefoundry/ir/hir/tensor/rank.py
src/tilefoundry/ir/hir/tensor/reduce.py
src/tilefoundry/ir/hir/tensor/repeat_interleave.py
src/tilefoundry/ir/hir/tensor/reshape.py
src/tilefoundry/ir/hir/tensor/shape_of.py
src/tilefoundry/ir/hir/tensor/slice.py
src/tilefoundry/ir/hir/tensor/split.py
src/tilefoundry/ir/hir/tensor/stack.py
src/tilefoundry/ir/hir/tensor/topk.py
src/tilefoundry/ir/hir/tensor/transpose.py
src/tilefoundry/ir/hir/tensor/tuple_get_item.py
src/tilefoundry/ir/hir/tensor/zeros.py
src/tilefoundry/ir/tir/__init__.py
src/tilefoundry/ir/tir/arith.py
src/tilefoundry/ir/tir/async_copy.py
src/tilefoundry/ir/tir/clamp.py
src/tilefoundry/ir/tir/dispatch.py
src/tilefoundry/ir/tir/intrinsic.py
src/tilefoundry/ir/tir/launch.py
src/tilefoundry/ir/tir/prim_function.py
src/tilefoundry/ir/tir/reduce.py
src/tilefoundry/ir/tir/shape.py
src/tilefoundry/ir/tir/stmt.py
src/tilefoundry/ir/tir/stmts.py
src/tilefoundry/ir/tir/symbol_ref.py
src/tilefoundry/ir/tir/sync.py
src/tilefoundry/ir/tir/verify.py
src/tilefoundry/ir/tir/cuda/__init__.py
src/tilefoundry/ir/tir/cuda/nn/__init__.py
src/tilefoundry/ir/tir/cuda/nn/mma.py
src/tilefoundry/ir/tir/cuda/nn/mma_atom.py
src/tilefoundry/ir/tir/memory/__init__.py
src/tilefoundry/ir/tir/memory/alloc_tensor.py
src/tilefoundry/ir/tir/memory/copy.py
src/tilefoundry/ir/tir/memory/fill.py
src/tilefoundry/ir/tir/memory/memory_span.py
src/tilefoundry/ir/tir/memory/ptr_of.py
src/tilefoundry/ir/tir/memory/tensor_view.py
src/tilefoundry/ir/tir/nn/__init__.py
src/tilefoundry/ir/tir/nn/relu.py
src/tilefoundry/ir/tir/nn/rms_norm.py
src/tilefoundry/ir/types/__init__.py
src/tilefoundry/ir/types/callable_type.py
src/tilefoundry/ir/types/dim.py
src/tilefoundry/ir/types/dim_typeinfer.py
src/tilefoundry/ir/types/dtype.py
src/tilefoundry/ir/types/shape_dim.py
src/tilefoundry/ir/types/shape_helpers.py
src/tilefoundry/ir/types/storage.py
src/tilefoundry/ir/types/substitute.py
src/tilefoundry/ir/types/tensor_type.py
src/tilefoundry/ir/types/utils.py
src/tilefoundry/ir/types/shard/__init__.py
src/tilefoundry/ir/types/shard/int_tuple.py
src/tilefoundry/ir/types/shard/layout.py
src/tilefoundry/ir/types/shard/layout_algebra.py
src/tilefoundry/ir/types/shard/mesh.py
src/tilefoundry/ir/types/shard/scope_match.py
src/tilefoundry/ir/types/shard/shard_layout.py
src/tilefoundry/ir/types/shard/utils.py
src/tilefoundry/parser/__init__.py
src/tilefoundry/parser/base.py
src/tilefoundry/parser/dispatch.py
src/tilefoundry/parser/hir_parser.py
src/tilefoundry/parser/overload.py
src/tilefoundry/parser/range_slice.py
src/tilefoundry/parser/static_eval.py
src/tilefoundry/parser/sugar.py
src/tilefoundry/parser/symtab.py
src/tilefoundry/parser/tir_parser.py
src/tilefoundry/passes/__init__.py
src/tilefoundry/passes/pass_base.py
src/tilefoundry/passes/pass_manager.py
src/tilefoundry/passes/transforms/__init__.py
src/tilefoundry/passes/transforms/bufferize.py
src/tilefoundry/passes/transforms/hir_to_tir.py
src/tilefoundry/passes/transforms/host_entry.py
src/tilefoundry/runtime/__init__.py
src/tilefoundry/runtime/decorator.py
src/tilefoundry/runtime/function.py
src/tilefoundry/runtime/loader.py
src/tilefoundry/runtime/measure.py
src/tilefoundry/runtime/module.py
src/tilefoundry/runtime/resource.py
src/tilefoundry/schedule/__init__.py
src/tilefoundry/schedule/api.py
src/tilefoundry/schedule/errors.py
src/tilefoundry/schedule/facts.py
src/tilefoundry/schedule/kernel_schedule.py
src/tilefoundry/schedule/plan.py
src/tilefoundry/schedule/registry.py
src/tilefoundry/schedule/render.py
src/tilefoundry/schedule/partition/__init__.py
src/tilefoundry/schedule/partition/facts.py
src/tilefoundry/schedule/partition/plan.py
src/tilefoundry/schedule/partition/problem.py
src/tilefoundry/schedule/partition/program.py
src/tilefoundry/schedule/partition/solve.py
src/tilefoundry/schedule/pipeline/__init__.py
src/tilefoundry/schedule/pipeline/facts.py
src/tilefoundry/schedule/pipeline/plan.py
src/tilefoundry/schedule/pipeline/problem.py
src/tilefoundry/schedule/pipeline/program.py
src/tilefoundry/schedule/pipeline/solve.py
src/tilefoundry/target/__init__.py
src/tilefoundry/target/base.py
src/tilefoundry/target/facts.py
src/tilefoundry/target/amx/__init__.py
src/tilefoundry/target/amx/architecture.py
src/tilefoundry/target/amx/atoms.py
src/tilefoundry/target/amx/device.py
src/tilefoundry/target/amx/facts.py
src/tilefoundry/target/amx/schedule.py
src/tilefoundry/target/amx/spec.py
src/tilefoundry/target/amx/target.py
src/tilefoundry/target/cuda/__init__.py
src/tilefoundry/target/cuda/architecture.py
src/tilefoundry/target/cuda/atoms.py
src/tilefoundry/target/cuda/device.py
src/tilefoundry/target/cuda/facts.py
src/tilefoundry/target/cuda/schedule.py
src/tilefoundry/target/cuda/spec.py
src/tilefoundry/target/cuda/target.py
src/tilefoundry/target/hardware/__init__.py
src/tilefoundry/target/hardware/apple_amx.toml
src/tilefoundry/target/hardware/apple_m2_pro.toml
src/tilefoundry/target/hardware/envelope.py
src/tilefoundry/target/hardware/nvidia_h200_sxm.toml
src/tilefoundry/target/hardware/nvidia_sm90.toml
src/tilefoundry/target/hardware/registry.py
src/tilefoundry/target/hardware/schema.py
src/tilefoundry/target/hardware/spec.py
src/tilefoundry/utils/__init__.py
src/tilefoundry/utils/markdown.py
src/tilefoundry/utils/spec_ref.py
src/tilefoundry/visitor_registry/__init__.py
src/tilefoundry/visitor_registry/access_relation.py
src/tilefoundry/visitor_registry/contexts.py
src/tilefoundry/visitor_registry/isl_utility.py
src/tilefoundry/visitor_registry/op_cost.py
src/tilefoundry/visitor_registry/registries.py
src/tilefoundry/visitor_registry/relation_build.py
src/tilefoundry/visitor_registry/shard_propagate.py
src/tilefoundry/visitor_registry/visitors.py
tests/__init__.py
tests/conftest.py
tests/analysis/__init__.py
tests/analysis/test_analysis_families.py
tests/analysis/test_analysis_invariants.py
tests/analysis/test_analyze_composition.py
tests/analysis/test_isl_utility.py
tests/analysis/test_poly_dynamic_shape.py
tests/analysis/test_poly_grid_region.py
tests/analysis/test_poly_nested_call.py
tests/analysis/test_poly_reshape.py
tests/analysis/test_poly_rmsnorm.py
tests/analysis/test_relation_build.py
tests/analysis/test_shard_propagate.py
tests/cli/test_cli.py
tests/cli/test_cli_check.py
tests/codegen/__init__.py
tests/codegen/test_dispatch_call_emit.py
tests/core/__init__.py
tests/core/test_analyze_at_a_size.py
tests/core/test_category_reexports.py
tests/core/test_dim_substitution.py
tests/core/test_gqa_online.py
tests/core/test_metadata.py
tests/core/test_module_ownership.py
tests/core/test_overload.py
tests/core/test_param_def.py
tests/core/test_pattern.py
tests/core/test_register_alias.py
tests/core/test_specialize.py
tests/core/test_typeinfer_discipline.py
tests/core/test_visitor_registry.py
tests/dsl/test_dsl_surface.py
tests/dsl/test_module_context.py
tests/dsl/test_module_decorator.py
tests/dump/__init__.py
tests/dump/test_dump.py
tests/e2e/__init__.py
tests/e2e/test_dynamic_cta_tir_handwritten.py
tests/e2e/test_dynamic_shape_dispatch.py
tests/e2e/test_host_launch.py
tests/e2e/test_mma_runtime.py
tests/e2e/test_mma_tir_handwritten.py
tests/e2e/test_rmsnorm_quant.py
tests/fixtures/__init__.py
tests/fixtures/demo_canonical.py
tests/fixtures/demo_ir.py
tests/fixtures/gqa_online.py
tests/fixtures/static_online.py
tests/inspection/__init__.py
tests/inspection/test_canonical_roundtrip.py
tests/inspection/test_module_tree_roundtrip.py
tests/inspection/test_python_printer.py
tests/inspection/test_roundtrip.py
tests/inspection/test_specialization_print.py
tests/integration/qwen3_1_7b_l3.py
tests/integration/installed/conftest.py
tests/integration/installed/smoke_analyze.py
tests/integration/installed/smoke_check.py
tests/integration/installed/smoke_inspect.py
tests/integration/installed/smoke_models.py
tests/integration/installed/smoke_schedule.py
tests/integration/installed/smoke_shipped.py
tests/integration/installed/smoke_spec.py
tests/integration/installed/smoke_tutorial.py
tests/ir/__init__.py
tests/ir/test_dim_var_envelope.py
tests/ir/test_dispatch_call.py
tests/ir/test_function_call_typeinfer.py
tests/ir/test_grid_region_dynamic_extent.py
tests/ir/test_grid_region_range_nested.py
tests/ir/test_shape_of.py
tests/ir/test_shard_layout_local_shape.py
tests/ir/test_simplify_dim.py
tests/ir/test_specialization_lowering.py
tests/ir/test_verify_letstmt.py
tests/ir/test_visitor.py
tests/ir_types/__init__.py
tests/ir_types/test_dtype.py
tests/ir_types/test_mma_fragment_layouts.py
tests/ir_types/test_shard_layout.py
tests/ir_types/test_storage.py
tests/ir_types/test_tensor_type.py
tests/models/__init__.py
tests/models/catalog.json
tests/models/corpus.py
tests/models/coverage_artifact.py
tests/models/decode_oracle.py
tests/models/dense_decode.py
tests/models/fixtures.py
tests/models/registry.py
tests/models/report.py
tests/models/test_analysis_is_hand_checkable.py
tests/models/test_analyze_coverage.py
tests/models/test_catalog.py
tests/models/test_cli_coverage.py
tests/models/test_corpus_harness.py
tests/models/test_corpus_integrity.py
tests/models/test_coverage_artifact.py
tests/models/test_max_context_coverage.py
tests/models/test_schedule_coverage.py
tests/models/test_sized_coverage.py
tests/models/test_tiled_mlp.py
tests/models/verified.json
tests/models/deepseek_v4_flash/__init__.py
tests/models/deepseek_v4_flash/case.py
tests/models/deepseek_v4_flash/config.json
tests/models/deepseek_v4_flash/hf_alias.py
tests/models/deepseek_v4_flash/model.py
tests/models/deepseek_v4_flash/reference.py
tests/models/deepseek_v4_flash/runtime.py
tests/models/deepseek_v4_flash/test_attention_decode.py
tests/models/deepseek_v4_flash/test_case.py
tests/models/deepseek_v4_flash/test_causal_lm_e2e.py
tests/models/deepseek_v4_flash/test_moe.py
tests/models/gemma2_2b/__init__.py
tests/models/gemma2_2b/case.py
tests/models/gemma2_2b/config.json
tests/models/gemma2_2b/hf_alias.py
tests/models/gemma2_2b/model.py
tests/models/gemma2_2b/reference.py
tests/models/gemma2_2b/test_decoder.py
tests/models/gemma2_2b/test_decoder_layer.py
tests/models/kimi_linear_48b_a3b/__init__.py
tests/models/kimi_linear_48b_a3b/case.py
tests/models/kimi_linear_48b_a3b/config.json
tests/models/kimi_linear_48b_a3b/model.py
tests/models/kimi_linear_48b_a3b/reference.py
tests/models/kimi_linear_48b_a3b/test_kda.py
tests/models/kimi_linear_48b_a3b/test_mla.py
tests/models/kimi_linear_48b_a3b/test_moe.py
tests/models/minicpm3_4b/__init__.py
tests/models/minicpm3_4b/case.py
tests/models/minicpm3_4b/config.json
tests/models/minicpm3_4b/hf_alias.py
tests/models/minicpm3_4b/model.py
tests/models/minicpm3_4b/reference.py
tests/models/minicpm3_4b/test_decoder.py
tests/models/minicpm3_4b/test_decoder_layer.py
tests/models/orchestrator/causal_lm/generation.py
tests/models/orchestrator/causal_lm/run.py
tests/models/qwen2_5_1_5b/__init__.py
tests/models/qwen2_5_1_5b/case.py
tests/models/qwen2_5_1_5b/config.json
tests/models/qwen2_5_1_5b/hf_alias.py
tests/models/qwen2_5_1_5b/model.py
tests/models/qwen2_5_1_5b/reference.py
tests/models/qwen2_5_1_5b/test_decoder.py
tests/models/qwen2_5_1_5b/test_decoder_layer.py
tests/models/qwen3_1_7b/__init__.py
tests/models/qwen3_1_7b/case.py
tests/models/qwen3_1_7b/config.json
tests/models/qwen3_1_7b/hf_alias.py
tests/models/qwen3_1_7b/model.py
tests/models/qwen3_1_7b/reference.py
tests/models/qwen3_1_7b/test_decoder.py
tests/models/qwen3_1_7b/test_decoder_layer.py
tests/models/qwen3_5_35b_a3b/__init__.py
tests/models/qwen3_5_35b_a3b/case.py
tests/models/qwen3_5_35b_a3b/config.json
tests/models/qwen3_5_35b_a3b/hf_alias.py
tests/models/qwen3_5_35b_a3b/model.py
tests/models/qwen3_5_35b_a3b/reference.py
tests/models/qwen3_5_35b_a3b/test_full_attention.py
tests/models/qwen3_5_35b_a3b/test_linear_attention.py
tests/models/qwen3_5_35b_a3b/test_moe.py
tests/models/qwen3_5_35b_a3b/test_provenance.py
tests/ops/__init__.py
tests/ops/eval_utils.py
tests/ops/test_argmax.py
tests/ops/test_async_copy.py
tests/ops/test_binary.py
tests/ops/test_cache_update.py
tests/ops/test_cast.py
tests/ops/test_clamp.py
tests/ops/test_concat.py
tests/ops/test_conv2d.py
tests/ops/test_eval_core.py
tests/ops/test_gather.py
tests/ops/test_insert_slice.py
tests/ops/test_layer_norm.py
tests/ops/test_matmul.py
tests/ops/test_mma.py
tests/ops/test_quant.py
tests/ops/test_reduce.py
tests/ops/test_relu.py
tests/ops/test_repeat_interleave.py
tests/ops/test_reshape.py
tests/ops/test_reshard.py
tests/ops/test_rms_norm.py
tests/ops/test_rope.py
tests/ops/test_sigmoid.py
tests/ops/test_silu.py
tests/ops/test_slice.py
tests/ops/test_softmax.py
tests/ops/test_softplus.py
tests/ops/test_sync.py
tests/ops/test_tanh.py
tests/ops/test_topk.py
tests/ops/test_transpose.py
tests/ops/test_unary.py
tests/ops/test_wide_load.py
tests/ops/typeinfer_utils.py
tests/parser/hir/__init__.py
tests/parser/hir/test_dim_arithmetic.py
tests/parser/hir/test_dsl_parse.py
tests/parser/hir/test_literal_storage.py
tests/parser/hir/test_nested_func_call.py
tests/parser/hir/test_parse_const_tensor.py
tests/parser/hir/test_parse_constraints.py
tests/parser/hir/test_parse_custom_op.py
tests/parser/hir/test_parse_errors.py
tests/parser/hir/test_parse_grid_region.py
tests/parser/hir/test_parse_shard_sugar.py
tests/parser/hir/test_parse_subscript.py
tests/parser/hir/test_parse_topology.py
tests/parser/hir/test_parse_tuple_unpack.py
tests/parser/hir/test_str_dtype_surface.py
tests/parser/tir/__init__.py
tests/parser/tir/test_dynamic_cta.py
tests/parser/tir/test_effect_form_selector.py
tests/parser/tir/test_mma_atom.py
tests/parser/tir/test_param_layout_sugar.py
tests/parser/tir/test_sync.py
tests/passes/__init__.py
tests/passes/test_bufferize.py
tests/passes/test_hir_to_tir.py
tests/passes/test_pass_manager.py
tests/runtime/__init__.py
tests/runtime/test_jit.py
tests/runtime/test_resource_device.py
tests/schedule/test_atom_facts.py
tests/schedule/test_partition.py
tests/schedule/test_pipeline.py
tests/schedule/test_schedule_api.py
tests/scripts/test_finalize_plan_context.py
tests/scripts/test_no_machine_paths_lint.py
tests/target/__init__.py
tests/target/test_amx_target.py
tests/target/test_hardware_resources.py
tests/target/test_target.py
tests/target/test_target_facts.py