Renamed k to ne

This commit is contained in:
Michael Wand 2026-03-24 01:44:45 -07:00
parent 53450f12f1
commit 7fd898beac
1 changed files with 2 additions and 2 deletions

View File

@ -621,12 +621,12 @@ template <typename dst_t>
static __global__ void dequantize_block_nvfp4(
const void * __restrict__ vx,
dst_t * __restrict__ yy,
const int64_t k) {
const int64_t ne) {
const int64_t i = blockIdx.x;
const int tid = threadIdx.x;
const int64_t base = i * QK_NVFP4;
if (base >= k) {
if (base >= ne) {
return;
}