Skip to content

Commit 35d16f1

Browse files
committed
error
1 parent c4e1db8 commit 35d16f1

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

paddle/fluid/inference/tensorrt/plugin/spmm_plugin.cu

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -673,7 +673,9 @@ nvinfer1::DataType SpmmPluginDynamic::getOutputDataType(
673673
PADDLE_ENFORCE_EQ(inputTypes[0] == nvinfer1::DataType::kFLOAT ||
674674
inputTypes[0] == nvinfer1::DataType::kHALF ||
675675
inputTypes[0] == nvinfer1::DataType::kINT8,
676-
true, "SpmmPluginDynamic is not support this format now");
676+
true,
677+
platform::errors::InvalidArgument(
678+
"SpmmPluginDynamic is not support this format now"));
677679

678680
return inputTypes[0];
679681
}

0 commit comments

Comments
 (0)