fix: typo

mickaelseznec · mickaelseznec · commit f2fab0b19f14 · 2025-07-22T16:32:11.000Z
Signed-off-by: Mickael Seznec &lt;mickael@mistral.ai&gt;
diff --git a/csrc/attention/mla/cutlass_mla_entry.cu b/csrc/attention/mla/cutlass_mla_entry.cu
@@ -35,7 +35,7 @@ void sm100_cutlass_mla_decode(
 
 int64_t sm100_cutlass_mla_get_workspace_size(int64_t max_seq_len,
                                              int64_t num_batches,
-                                             int64_t sm_count = 0,
+                                             int64_t sm_count,
                                              int64_t num_kv_splits) {
   TORCH_CHECK_NOT_IMPLEMENTED(false, "No compiled cutlass MLA");
 }

Original file line number	Diff line number	Diff line change
`@@ -35,7 +35,7 @@ void sm100_cutlass_mla_decode(`
`35`	`35`
`36`	`36`	`int64_t sm100_cutlass_mla_get_workspace_size(int64_t max_seq_len,`
`37`	`37`	`int64_t num_batches,`
`38`		`- int64_t sm_count = 0,`
	`38`	`+ int64_t sm_count,`
`39`	`39`	`int64_t num_kv_splits) {`
`40`	`40`	`TORCH_CHECK_NOT_IMPLEMENTED(false, "No compiled cutlass MLA");`
`41`	`41`	`}`