Move Dygraph Indexing Parsing to Pybind by zoooo0820 · Pull Request #58643

Move Dygraph Indexing Parsing to Pybind by zoooo0820 · Pull Request #58643 · PaddlePaddle/Paddle

added 7 commits

October 18, 2023 03:14

…th from stack to unsqueeze

zoooo0820 changed the title ~~[Just for Test] Indexing to pybind~~ Move Dygraph Indexing Parsing to Pybind

Nov 17, 2023

SecretXV pushed a commit to SecretXV/Paddle that referenced this pull request

Nov 28, 2023

* start down to pybind

* basic-getitem down

* fix compile error; basic-getitem seems work fine

* finish advanced-getitem, runs well but still has bug to fix

* fix parsing bug; getitem seems ok

* finish setitem, still has bug to fix

* fix some bug in advanced setitem

* fix CI error about bool-tensor indexing

* fix advance-indexing ci error

* remove transpose if there is no need

* support single py_boolean in index

* add assert densetensor to avoid error in auto-parallel mode

* add fast path for case single int tensor in getitem

* change set_value_dygraph_function to ad_func

* change vector type and fix the set_value_grad ut

* fix inplace version error in backward by assign

* remove any-check in bool tensor to boost getitem

* remove no-used python code, keep xpu code to fix; py-code add fast path from stack to unsqueeze

* fix py_bool cuda error

* fix cuda config error when gpu_memory_available after all-false bool-tensor indexing

* fix ci in 0-size gather_nd indexing

* add dist tensor check