Skip to content

【PaddlePaddle Hackathon】28、Paddle TensorRT 增加gru的支持 #35979

Closed
@TCChenlong

Description

@TCChenlong

(此 ISSUE 为 PaddlePaddle Hackathon 活动的任务 ISSUE,更多详见PaddlePaddle Hackathon

【任务说明】

  • 任务标题:Paddle TensorRT 增加gru的支持

  • 技术标签:预测,C++,TensorRT

  • 任务难度:中等

  • 详细描述:Paddle Inference通过子图接入的方式,在SimpleOpTypeSetTeller里判断一个OP是否可以运行到TensorRT中,支持的OP经过对应的OpConverter转变成TensorRT的原生Layer或者Plugin。通过增加该OP的支持,可以助于熟悉Paddle Inference、TensorRT等深度学习推理框架的典型执行流程。加入方式可以参考reduce_sum的实现,TensorRT中GRU的实现参考TensorRT work-with-loops

【提交内容】

  • 任务 PR 到 Paddle

  • 相关技术文档

  • 任务单测文件(静态shape/动态shape/FP32/FP16)

【技术要求】

  • 熟练掌握 C++

  • 良好的编程规范

  • 熟悉TensorRT的使用和paddle推理框架

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions