ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-20 06:49:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Revert "[CK-Tile] Add the API to load SGPR (#2878)" (#2904)

Browse Source 
This reverts commit 2cbbf5dcb3.
This commit is contained in:
asleepzzz
2025-09-24 05:33:51 +08:00
committed by GitHub
parent 959df2a155
commit f161b5b738
40 changed files with 167 additions and 273 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
include/ck_tile/ops/topk_softmax/kernel/topk_softmax_kernel.hpp
View File

@@ -96,9 +96,9 @@ struct TopkSoftmaxKernel
if(block_row_id > kargs.num_rows)
return;
index_t block_os_inp = amd_wave_read_first_lane(block_row_id * kargs.stride_input);
index_t block_os_out = amd_wave_read_first_lane(block_row_id * kargs.stride_output);
index_t num_rows_rem = amd_wave_read_first_lane(kargs.num_rows - block_row_id);
index_t block_os_inp = __builtin_amdgcn_readfirstlane(block_row_id * kargs.stride_input);
index_t block_os_out = __builtin_amdgcn_readfirstlane(block_row_id * kargs.stride_output);
index_t num_rows_rem = __builtin_amdgcn_readfirstlane(kargs.num_rows - block_row_id);
const auto input_window = [&]() {
const InputType* p_input =