Skip to content

Commit 7bd8572

Browse files
committed
rename
Signed-off-by: leslie-fang25 <[email protected]>
1 parent b2b6c19 commit 7bd8572

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

cpp/tensorrt_llm/kernels/quantization.cu

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -146,12 +146,12 @@ void invokeFP4Quantization(int b, int m, int n, T const* input, float const* SFS
146146
int const numBlocksPerSM = std::max(1u, 2048u / block.x);
147147
// The number of blocks for m. The m dimension will be padded to 128 for swizzled layout.
148148
int numBlocksForM = layout == QuantizationSFLayout::SWIZZLED ? PadUpFn(m, 128) : m;
149-
int grid_size = std::min(numBlocksForM, multiProcessorCount * numBlocksPerSM);
150-
if (grid_size == 0)
149+
int gridSize = std::min(numBlocksForM, multiProcessorCount * numBlocksPerSM);
150+
if (gridSize == 0)
151151
{
152-
grid_size = 1;
152+
gridSize = 1;
153153
}
154-
dim3 grid(grid_size);
154+
dim3 grid(gridSize);
155155

156156
// Launch the cvt kernel.
157157
auto* kernel_instance = useUE8M0
@@ -170,12 +170,12 @@ void invokeFP4Quantization(int b, int m, int n, T const* input, float const* SFS
170170
int const numBlocksPerSM = std::max(1u, 2048u / block.x);
171171
// The number of blocks for m. The m dimension will be padded to 128 for swizzled layout.
172172
int numBlocksForM = layout == QuantizationSFLayout::SWIZZLED ? PadUpFn(m, 128) : m;
173-
int grid_size = std::min(numBlocksForM, multiProcessorCount * numBlocksPerSM);
174-
if (grid_size == 0)
173+
int gridSize = std::min(numBlocksForM, multiProcessorCount * numBlocksPerSM);
174+
if (gridSize == 0)
175175
{
176-
grid_size = 1;
176+
gridSize = 1;
177177
}
178-
dim3 grid(grid_size);
178+
dim3 grid(gridSize);
179179

180180
// Launch the cvt kernel.
181181
auto* kernel_instance = useUE8M0

0 commit comments

Comments
 (0)