Skip to content

Commit

Permalink
Update stack_op_plugin.cu
Browse files Browse the repository at this point in the history
comments
  • Loading branch information
b3602sss authored Dec 3, 2021
1 parent 48ee7da commit dff5a94
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion paddle/fluid/inference/tensorrt/plugin/stack_op_plugin.cu
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ bool StackPluginDynamic::supportsFormatCombination(
if (with_fp16_) {
return (
// It's workaround for ernie fix len model.
// Enable float, half on the same time leaads to trt hang.
// Enabling float, half on the same time will cause trt hang.
#if IS_TRT_VERSION_LT(8000)
in.type == nvinfer1::DataType::kFLOAT ||
#endif
Expand Down

1 comment on commit dff5a94

@paddle-bot-old
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Congratulation! Your pull request passed all required CI. You could ask reviewer(s) to approve and merge. 🎉

Please sign in to comment.