Paddle
Paddle copied to clipboard
Optimize topk's performance when k is small and input_width is large
PR types
Performance optimization
PR changes
OPs
Describe
Optimize topk's performance when k is small and input_width is large
dtype:FP32,循环测试10w次,取平均值。
优化后在k值较小且input_width较大时速度提升3-4倍。
你的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.