Fix inconsistency between index_fill and index_fill_ #59863
Conversation
|
你的PR提交成功，感谢你对开源项目的贡献! |
|
❌ The PR is not created using PR's template. You can refer to this Demo. |
| @@ -1621,7 +1621,7 @@ class TestDygraphInplaceIndexFill(TestDygraphInplace): | |||
| def init_data(self): | |||
| self.input_var_numpy = np.random.random((20, 40)) | |||
| self.dtype = "float32" | |||
| self.axis = 0 | |||
| self.axis = 1 | |||
There was a problem hiding this comment.
这个修改是否是必要的呢，目前axis=0是否表现正常？
There was a problem hiding this comment.
axis=0是正常的，axis=1覆盖情况更广，之前就是因为axis=0导致没有测试到这个API的inplace版本有不一致问题
| if inplace: | ||
| paddle.transpose(x, perm) |
There was a problem hiding this comment.
目前有inplace版本paddle.transpose_
此外，不建议在该API中使用索引语法实现赋值，会带来解析索引的耗时，此前使用transpose + index_put的逻辑，本质上是已经根据该场景解析过后决定了要调用哪些API
There was a problem hiding this comment.
1.目前paddle文档上没有paddle.transpose_
2.使用paddle.transpose_实现会有inplace_version跳变问题，inplace_version貌似在python端只能增不能减，所以我更改了一部分version相关测试
test/legacy_test/test_inplace.py
Outdated
| self.assertEqual(var.inplace_version, 7) | ||
|
|
||
| def test_backward_error(self): | ||
| pass |
There was a problem hiding this comment.
一样是inplace_version的问题，已添加
|
@zoooo0820 麻烦确认下是否可以合入，https://github.com/PaddlePaddle/PaConvert/ 中需要增加这个API的单测使用 |
|
@Patrick-Star125 |
|
@co63oc 该PR已合入 |
PR types
Bug fixes
PR changes
APIs
Description
我之前错误的认为transpose是inplace的导致两个API最终结果不相等，这里将transpose去除，直接对x赋值
会增加一些的赋值成本，但是transpose没有inplace版本，目前paddle也没有可以替代实现交换轴的inplace API
Link
相关PR：#57416