[Compile] Fix nvfp4 compile warning (#38573)

Signed-off-by: yewentao256 <zhyanwentao@126.com>

[Compile] Fix nvfp4 compile warning (#38573)
Signed-off-by: yewentao256 <zhyanwentao@126.com>
c9a9db0e · Wentao Ye · GitHub · cbe7d180 · c9a9db0e
Unverified Commit c9a9db0e authored Apr 01, 2026 by Wentao Ye Committed by GitHub Apr 01, 2026
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 4 deletions

csrc/libtorch_stable/quantization/fp4/nvfp4_quant_entry.cu csrc/libtorch_stable/quantization/fp4/nvfp4_quant_entry.cu +7 -4

No files found.
--- a/csrc/libtorch_stable/quantization/fp4/nvfp4_quant_entry.cu
+++ b/csrc/libtorch_stable/quantization/fp4/nvfp4_quant_entry.cu
@@ -58,16 +58,19 @@ void silu_and_mul_scaled_fp4_experts_quant_sm1xxa(
    torch::stable::Tensor const& output_scale_offset_by_experts);
 #endif

+#if (defined(ENABLE_NVFP4_SM100) && ENABLE_NVFP4_SM100) || \
+    (defined(ENABLE_NVFP4_SM120) && ENABLE_NVFP4_SM120)
 static bool nvfp4_quant_sm_supported() {
  const int32_t sm = get_sm_version_num();
-#if defined(ENABLE_NVFP4_SM100) && ENABLE_NVFP4_SM100
+  #if defined(ENABLE_NVFP4_SM100) && ENABLE_NVFP4_SM100
  if (sm >= 100 && sm < 120) return true;
-#endif
-#if defined(ENABLE_NVFP4_SM120) && ENABLE_NVFP4_SM120
+  #endif
+  #if defined(ENABLE_NVFP4_SM120) && ENABLE_NVFP4_SM120
  if (sm >= 120 && sm < 130) return true;
-#endif
+  #endif
  return false;
 }
+#endif

 void scaled_fp4_quant_out(torch::stable::Tensor const& input,
                          torch::stable::Tensor const& input_sf,