We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c4e1db8 commit 35d16f1Copy full SHA for 35d16f1
paddle/fluid/inference/tensorrt/plugin/spmm_plugin.cu
@@ -673,7 +673,9 @@ nvinfer1::DataType SpmmPluginDynamic::getOutputDataType(
673
PADDLE_ENFORCE_EQ(inputTypes[0] == nvinfer1::DataType::kFLOAT ||
674
inputTypes[0] == nvinfer1::DataType::kHALF ||
675
inputTypes[0] == nvinfer1::DataType::kINT8,
676
- true, "SpmmPluginDynamic is not support this format now");
+ true,
677
+ platform::errors::InvalidArgument(
678
+ "SpmmPluginDynamic is not support this format now"));
679
680
return inputTypes[0];
681
}
0 commit comments