File tree Expand file tree Collapse file tree 1 file changed +5
-4
lines changed
csrc/fused_moe/cutlass_backend Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -822,11 +822,12 @@ class FusedMoeRunner : public tvm::ffi::ModuleObj {
822822 auto const fc2_dequant = quant_scales.value ()[2 ];
823823 auto const fc1_input_dequant = quant_scales.value ()[3 ];
824824
825- TVM_FFI_ICHECK (fc1_dequant.get () != nullptr ) << " Expecting fc1_dequant to be non null" ;
826- TVM_FFI_ICHECK (fc2_quant.get () != nullptr ) << " Expecting fc2_quant to be non null" ;
827- TVM_FFI_ICHECK (fc2_dequant.get () != nullptr )
825+ TVM_FFI_ICHECK (fc1_dequant.GetDLTensorPtr () != nullptr )
826+ << " Expecting fc1_dequant to be non null" ;
827+ TVM_FFI_ICHECK (fc2_quant.GetDLTensorPtr () != nullptr ) << " Expecting fc2_quant to be non null" ;
828+ TVM_FFI_ICHECK (fc2_dequant.GetDLTensorPtr () != nullptr )
828829 << " Expecting fc2_dequant_dequant to be non null" ;
829- TVM_FFI_ICHECK (fc1_input_dequant.get () != nullptr )
830+ TVM_FFI_ICHECK (fc1_input_dequant.GetDLTensorPtr () != nullptr )
830831 << " Expecting fc1_input_dequant to be non null" ;
831832
832833 // Check types
You can’t perform that action at this time.
0 commit comments