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 b131f6916b..5bef6f8719 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 @@ -822,11 +822,12 @@ 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) + TVM_FFI_ICHECK(fc1_dequant.GetDLTensorPtr() != nullptr) + << "Expecting fc1_dequant to be non null"; + TVM_FFI_ICHECK(fc2_quant.GetDLTensorPtr() != nullptr) << "Expecting fc2_quant to be non null"; + TVM_FFI_ICHECK(fc2_dequant.GetDLTensorPtr() != nullptr) + << "Expecting fc2_dequant to be non null"; + TVM_FFI_ICHECK(fc1_input_dequant.GetDLTensorPtr() != nullptr) << "Expecting fc1_input_dequant to be non null"; // Check types