[Macro] Increase macro constant MAX_RANK_SUPPORTED

[HydrogenSulfate]

PR Category

Operator Mechanism

PR Types

Improvements

Description

prod_grad composite OP will fail when given input x.ndim>6, because prod_grad calls expand OP in its' implementation. So a fine solution is increasing MAX_RANK_SUPPORTED from 6 to 8.

after modification, 8D unitests can be passed successfully as below:

$ python test/legacy_test/test_reduce_op.py

import paddle

paddle.framework.core.set_prim_eager_enabled(True)

x = paddle.randn([1, 2, 1, 1, 2, 1, 1, 1])
x.stop_gradient = False
y = paddle.prod(x)

print(x)
g = paddle.grad(y, x)[0]
print(g)

'''
Tensor(shape=[1, 2, 1, 1, 1, 1, 1, 1], dtype=float32, place=Place(gpu:0), stop_gradient=False,
       [[[[[[[[0.04061465]]]]]],





         [[[[[[0.18230575]]]]]]]])
Tensor(shape=[1, 2, 1, 1, 1, 1, 1, 1], dtype=float32, place=Place(gpu:0), stop_gradient=True,
       [[[[[[[[0.18230575]]]]]],





         [[[[[[0.04061465]]]]]]]])
'''

[paddle-bot]

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

[paddle-bot]

❌ The PR is not created using PR's template. You can refer to this Demo.
Please use PR's template, it helps save our maintainers' time so that more developers get helped.

[zyfncg]

LGTM