SIGN IN SIGN UP

[rocm-libraries] ROCm/rocm-libraries#8086 (commit 33e2420)

fix(ck): Stream-K Tile Engine GPU Query Fix

## Motivation
The Stream-K tile engine validation utilities call rocminfo at build
time to detect the GPU architecture, which prevents building on CPU-only
nodes. The GPU target is already known from CMake's
SUPPORTED_GPU_TARGETS, so runtime hardware detection is unnecessary
during code generation.

## Technical Details
The gemm_streamk_validation_utils.py file used
subprocess.check_output(["rocminfo"]) to query GPU hardware at CMake
configure time. This call originated from get_gpu_name_by_id() and used
during tile configuration validation. On CPU-only build nodes, this
fails because rocminfo either doesn't exist or returns no GPU devices.
The main changes are as follows:

- Removed runtime GPU detection infrastructure: deleted
get_gpu_name_by_id(), set_gpu_targets(), get_configured_gpu_targets(),
the _configured_gpu_targets module variable, and the GPU_NAME_PATTERN
regex.
- Added gpu_target as an explicit parameter. is_tile_config_valid(),
validate_gemm(), validate_warp_tile_combination(),
validate_warp_configuration(), and validate_lds_capacity() now accept
gpu_target as a required parameter instead of querying hardware
internally. The corresponding changes were also made in
gemm_streamk_instance_builder.py and CMakeLists.txt
- Added WARP_SUPPORTED_COMBINATIONS for per-GPU warp config validation,
and LDS_SIZE_MAP / DEFAULT_LDS_SIZE for GPU-aware LDS capacity checks.

## Test Plan
The benchmarks were compiled and run on a GPU as well as a CPU only node
to verify correctness.

## Test Result
All tests passed

## Related
JIRA ID : AICK-1635

## Submission Checklist
- [x] Look over the contributing guidelines at
https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.
A
arai713 committed
cf295d9def7bedee915ce2810f0dc594f6a9dc02
Parent: 5a74dec
Committed by assistant-librarian[bot] <assistant-librarian[bot]@users.noreply.github.com> on 7/24/2026, 5:29:35 PM