Ext: Fix register call for float

ROCm doesn't support the register keyword. This should fix compile.

Signed-off-by: kingbri <8082010+kingbri1@users.noreply.github.com>
This commit is contained in:
kingbri
2025-05-12 14:15:56 -04:00
parent 0a7733110e
commit bb4206d5bc

View File

@@ -279,7 +279,11 @@ __global__ void softmax128_topk_norm_kernel
const int row = blockIdx.y * WARPSIZE + threadIdx.x;
if (row >= rows) return;
#if defined(USE_ROCM)
float f[128];
#else
register float f[128];
#endif
int4* row_ptr = reinterpret_cast<int4*>(x + row * 128);