[CINN] Replace unit vars to zero before resizing buffer

PaddlePaddle / Paddle

PArallel Distributed Deep LEarning: Machine Learning Framework from Industrial Practice （『飞桨』核心框架，深度学习&机器学习高性能单机、分布式训练和跨平台部署）

http://www.paddlepaddle.org/

Apache License 2.0

22.28k stars 5.61k forks source link

[CINN] Replace unit vars to zero before resizing buffer #69608

Closed lshpku closed 16 hours ago

lshpku commented 1 day ago

PR Category

CINN

PR Types

Bug fixes

Description

In OptimizeExprGPU, move the ReplaceUnitVarToZero pass before the EliminateCommonFactorOfLocalIndex pass. Otherwise, we may fail to eliminate some unit variables (variables that are in range [0, 1)) in local buffer's indices when there is transpose in the kernel.

Pcard-85711

paddle-bot[bot] commented 1 day ago

你的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.