composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-10 16:28:38 +00:00

Files

Thrupti Raj Lakshmana Gowda c31fc4df52 [rocm-libraries] ROCm/rocm-libraries#7311 (commit 79d8cae)

[CK Tile Engine] Daily tier sampling for tile engine GEMM  (#7311)

Summary
- Replace uniform random instance sampling (random.shuffle) with
scrambled Sobol + Latin Hypercube + maximin space-filling
sampling, per the Tile Engine Benchmark Sampling RFC
- Add op-weighted budget allocation via new
TILE_ENGINE_SAMPLING_TIER=daily CMake knob that auto-distributes 8,000
instances across
ops proportional to registered weights in op_weights.json
  - Emit chosen_instances.json manifests for reproducibility tracking
- Consolidate 5 copies of sampling logic into single _apply_sampling()
method on the base class
Jenkinsfile changes
Replace per-op -D *_MAX_INSTANCES=250 with single -D
TILE_ENGINE_SAMPLING_TIER=daily in gfx942/gfx950/gfx1201 stages. Budget
  auto-distributes (8000 total per GPU target).

---------

Co-authored-by: Claude Sonnet 4 <noreply@anthropic.com>

2026-05-21 02:17:42 -05:00

__init__.py

[rocm-libraries] ROCm/rocm-libraries#7311 (commit 79d8cae)

2026-05-21 02:17:42 -05:00

allocate_budget.py

[rocm-libraries] ROCm/rocm-libraries#7311 (commit 79d8cae)

2026-05-21 02:17:42 -05:00

budget.py

[rocm-libraries] ROCm/rocm-libraries#7311 (commit 79d8cae)

2026-05-21 02:17:42 -05:00

feasible_set.py

[rocm-libraries] ROCm/rocm-libraries#7311 (commit 79d8cae)