Restrict <sm130

mgoin · web-flow · commit 51bcb3ba628e · 2025-12-01T15:27:42.000-05:00
Signed-off-by: Michael Goin &lt;mgoin64@gmail.com&gt;
diff --git a/csrc/quantization/fp4/nvfp4_scaled_mm_entry.cu b/csrc/quantization/fp4/nvfp4_scaled_mm_entry.cu
@@ -53,7 +53,7 @@ void cutlass_scaled_fp4_mm(torch::Tensor& D,
 #endif
 
 #if defined(ENABLE_NVFP4_SM120) && ENABLE_NVFP4_SM120
-  if (sm >= 120) {
+  if (sm >= 120 && sm < 130) {
     cutlass_scaled_fp4_mm_sm120a(D, A, B, A_sf, B_sf, alpha);
     return;
   }

Original file line number	Diff line number	Diff line change
`@@ -53,7 +53,7 @@ void cutlass_scaled_fp4_mm(torch::Tensor& D,`
`53`	`53`	`#endif`
`54`	`54`
`55`	`55`	`#if defined(ENABLE_NVFP4_SM120) && ENABLE_NVFP4_SM120`
`56`		`- if (sm >= 120) {`
	`56`	`+ if (sm >= 120 && sm < 130) {`
`57`	`57`	`cutlass_scaled_fp4_mm_sm120a(D, A, B, A_sf, B_sf, alpha);`
`58`	`58`	`return;`
`59`	`59`	`}`