Skip to content

GitLab

Unverified Commit c9a9db0e authored by Wentao Ye's avatar Wentao Ye Committed by GitHub
Browse files

[Compile] Fix nvfp4 compile warning (#38573) 


Signed-off-by: default avataryewentao256 <zhyanwentao@126.com>
parent cbe7d180
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 4 deletions
csrc/libtorch_stable/quantization/fp4/nvfp4_quant_entry.cu
View file @ c9a9db0e
...@@ -58,16 +58,19 @@ void silu_and_mul_scaled_fp4_experts_quant_sm1xxa( ...@@ -58,16 +58,19 @@ void silu_and_mul_scaled_fp4_experts_quant_sm1xxa(
torch::stable::Tensor const& output_scale_offset_by_experts); torch::stable::Tensor const& output_scale_offset_by_experts);
#endif #endif
#if (defined(ENABLE_NVFP4_SM100) && ENABLE_NVFP4_SM100) || \
(defined(ENABLE_NVFP4_SM120) && ENABLE_NVFP4_SM120)
static bool nvfp4_quant_sm_supported() { static bool nvfp4_quant_sm_supported() {
const int32_t sm = get_sm_version_num(); 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; if (sm >= 100 && sm < 120) return true;
#endif #endif
#if defined(ENABLE_NVFP4_SM120) && ENABLE_NVFP4_SM120 #if defined(ENABLE_NVFP4_SM120) && ENABLE_NVFP4_SM120
if (sm >= 120 && sm < 130) return true; if (sm >= 120 && sm < 130) return true;
#endif #endif
return false; return false;
} }
#endif
void scaled_fp4_quant_out(torch::stable::Tensor const& input, void scaled_fp4_quant_out(torch::stable::Tensor const& input,
torch::stable::Tensor const& input_sf, torch::stable::Tensor const& input_sf,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment