add cumsum prim backward #50565
add cumsum prim backward #50565
Conversation
|
你的PR提交成功，感谢你对开源项目的贡献! |
|
由 #49518 引入的bug: |
see https://github.com/PaddlePaddle/CINN/blob/develop/cinn/frontend/op_mappers/paddle/cumsum.cc Compared with the paddle cumsum operator, CINN's cumsum operator lacks some parameters, and it cannot be aligned when connected to CINN. Consider adding operator mapping or enhancing the CINN cumsum operator in the future. |
|
|
||
| return res | ||
|
|
||
| # def test_cinn(self): |
There was a problem hiding this comment.
leave to do here and remove this test, no commented code should be pushed.
… add_cumsum_grad_for_prim
… add_cumsum_grad_for_prim
| self.attrs = {'axis': 2} | ||
| self.inputs = {'X': np.random.random((5, 6, 10)).astype("float64")} | ||
| self.outputs = {'Out': self.inputs['X'].cumsum(axis=2)} | ||
|
|
||
| def test_check_output(self): | ||
| self.check_output() | ||
| self.check_output(check_prim=True) |
There was a problem hiding this comment.
基础算子没有前向拆分规则，不需要在前向测试中加上组合规则测试，可以去掉这个参数。
| @@ -138,35 +144,41 @@ def setUp(self): | |||
| } | |||
|
|
|||
| def test_check_output(self): | |||
| self.check_output() | |||
| self.check_output(check_prim=True) | |||
| self.attrs = {'axis': 1} | ||
| self.inputs = {'X': np.random.random((5, 6, 10)).astype("float64")} | ||
| self.outputs = {'Out': self.inputs['X'].cumsum(axis=1)} | ||
|
|
||
| def test_check_output(self): | ||
| self.check_output() | ||
| self.check_output(check_prim=True) |
| self.attrs = {'axis': 0} | ||
| self.inputs = {'X': np.random.random((5, 6, 10)).astype("float64")} | ||
| self.outputs = {'Out': self.inputs['X'].cumsum(axis=0)} | ||
|
|
||
| def test_check_output(self): | ||
| self.check_output() | ||
| self.check_output(check_prim=True) |
| self.inputs = {'X': np.random.random((5, 20)).astype("float64")} | ||
| self.outputs = {'Out': self.inputs['X'].cumsum(axis=1)} | ||
|
|
||
| def test_check_output(self): | ||
| self.check_output() | ||
| self.check_output(check_prim=True) |
paddle/phi/api/yaml/op_compat.yaml
Outdated
| attrs : | ||
| axis : axis |
| bool flatten = static_cast<bool>(this->Attr<bool>("flatten")); | ||
| bool exclusive = static_cast<bool>(this->Attr<bool>("exclusive")); | ||
| bool reverse = static_cast<bool>(this->Attr<bool>("reverse")); | ||
| VLOG(6) << "Runing add_grad composite func"; |
PR types
Others
PR changes
OPs
Describe
add cumsum prim backward