Skip to content

Commit ac93024

Browse files
DefTruthlulmer
authored andcommitted
[Bugfix][W8A8] fixed cutlass block fp8 binding (vllm-project#14796)
Signed-off-by: Louis Ulmer <[email protected]>
1 parent dd372c5 commit ac93024

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

csrc/torch_bindings.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -370,7 +370,7 @@ TORCH_LIBRARY_EXPAND(TORCH_EXTENSION_NAME, ops) {
370370
"cutlass_scaled_mm_supports_block_fp8(int cuda_device_capability) -> "
371371
"bool");
372372
ops.impl("cutlass_scaled_mm_supports_block_fp8",
373-
&cutlass_scaled_mm_supports_fp8);
373+
&cutlass_scaled_mm_supports_block_fp8);
374374

375375
// Check if cutlass sparse scaled_mm is supported for CUDA devices of the
376376
// given capability

0 commit comments

Comments
 (0)