Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

paddle-TRT support float64 #55520

Merged
merged 4 commits into from
Jul 27, 2023
Merged

Conversation

ming1753
Copy link
Contributor

@ming1753 ming1753 commented Jul 18, 2023

PR types

Others

PR changes

Others

Description

Pcard-71501

  1. 仿照对int64的支持方式新增了paddle-TRT对float64类型的支持功能
  • 子图的输入为float64时插入cast转换为float32,输出为float64时插cast由float32转回
  1. fill_any_like op支持int64
  • 在fill_any_like op coverter中将op_desc(dtype 3)(int64) 视作(dtype 2)(int 32)

@paddle-bot
Copy link

paddle-bot bot commented Jul 18, 2023

你的PR提交成功,感谢你对开源项目的贡献!
请关注后续CI自动化测试结果,详情请参考Paddle-CI手册
Your PR has been submitted. Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

Copy link
Contributor

@Wangzheee Wangzheee left a comment

Choose a reason for hiding this comment

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

LGTM

@Wangzheee Wangzheee merged commit 8b06303 into PaddlePaddle:develop Jul 27, 2023
27 checks passed
wz1qqx pushed a commit to wz1qqx/Paddle that referenced this pull request Jul 31, 2023
* Paddle-TRT support float64  in/out type, support fill_any_like_op in int64
jinjidejinmuyan pushed a commit to jinjidejinmuyan/Paddle that referenced this pull request Aug 30, 2023
* Paddle-TRT support float64  in/out type, support fill_any_like_op in int64
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants