[CINN] Add infer_symbol_shape for memory_efficient_attention

PaddlePaddle / Paddle

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

http://www.paddlepaddle.org/

Apache License 2.0

21.66k stars 5.44k forks source link

[CINN] Add infer_symbol_shape for memory_efficient_attention #63999

Closed zyfncg closed 2 weeks ago

zyfncg commented 2 weeks ago

PR Category

CINN

PR Types

Devs

Description

Pcard-67164

memory_efficient_attention算子添加InferSymbolicShape函数。

paddle-bot[bot] commented 2 weeks 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.