diff --git a/csrc/fused_moe/cutlass_backend/flashinfer_cutlass_fused_moe_sm100_binding.cu b/csrc/fused_moe/cutlass_backend/flashinfer_cutlass_fused_moe_sm100_binding.cu index 298242acf4..e14132c241 100644 --- a/csrc/fused_moe/cutlass_backend/flashinfer_cutlass_fused_moe_sm100_binding.cu +++ b/csrc/fused_moe/cutlass_backend/flashinfer_cutlass_fused_moe_sm100_binding.cu @@ -800,6 +800,13 @@ class FusedMoeRunner : public tvm::ffi::ModuleObj { auto const fc2_dequant = quant_scales.value()[2]; auto const fc1_input_dequant = quant_scales.value()[3]; + TVM_FFI_ICHECK(fc1_dequant.get() != nullptr) << "Expecting fc1_dequant to be non null"; + TVM_FFI_ICHECK(fc2_quant.get() != nullptr) << "Expecting fc2_quant to be non null"; + TVM_FFI_ICHECK(fc2_dequant.get() != nullptr) + << "Expecting fc2_dequant_dequant to be non null"; + TVM_FFI_ICHECK(fc1_input_dequant.get() != nullptr) + << "Expecting fc1_input_dequant to be non null"; + // Check types CHECK_INPUT_TYPE(fc1_dequant, dl_float32); CHECK_INPUT_TYPE(fc2_quant, dl_float32);