Skip to content

Commit 83cea35

Browse files
committed
fix build error
Signed-off-by: Lain <[email protected]>
1 parent 59d5e36 commit 83cea35

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

csrc/quantization/cutlass_w8a8/c3x/scaled_mm_blockwise_sm100_fp8_dispatch.cuh

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -204,10 +204,7 @@ void cutlass_gemm_blockwise_sm100_fp8_dispatch(torch::Tensor& out,
204204
torch::Tensor const& b,
205205
torch::Tensor const& a_scales,
206206
torch::Tensor const& b_scales) {
207-
auto m = a.size(0);
208-
auto k = a.size(1);
209-
auto n = b.size(1);
210-
int sms;
207+
int32_t m = a.size(0), n = b.size(1), k = a.size(1), sms;
211208
cudaDeviceGetAttribute(&sms, cudaDevAttrMultiProcessorCount, a.get_device());
212209

213210
constexpr int TILE_K = 128;

0 commit comments

Comments
 (0)