mirror of
https://github.com/turboderp-org/exllamav2.git
synced 2026-03-15 00:07:26 +00:00
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:
@@ -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);
|
||||
|
||||
|
||||
Reference in New Issue
Block a user