mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-06-28 18:56:59 +00:00
c31fc4df52
[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>
20 lines
672 B
CMake
20 lines
672 B
CMake
# Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
include_directories(BEFORE
|
|
${CMAKE_CURRENT_LIST_DIR}/include
|
|
${CMAKE_CURRENT_LIST_DIR}/ops
|
|
)
|
|
|
|
set(TILE_ENGINE_SAMPLING_TIER "" CACHE STRING
|
|
"Sampling tier: 'daily' (8000 budget) or integer budget (empty = no cap)")
|
|
set(TILE_ENGINE_SAMPLING_SEED "" CACHE STRING
|
|
"Override sampling seed (empty = daily rotation)")
|
|
|
|
add_subdirectory(ops/fmha EXCLUDE_FROM_ALL)
|
|
add_subdirectory(ops/gemm EXCLUDE_FROM_ALL)
|
|
add_subdirectory(ops/gemm_streamk EXCLUDE_FROM_ALL)
|
|
add_subdirectory(ops/pooling EXCLUDE_FROM_ALL)
|
|
add_subdirectory(ops/reduce EXCLUDE_FROM_ALL)
|
|
|