POC of dynamicShape+functionalization integration (#4611)
* Support dynamism on "transpose", unsqueeze_copy, and view_copy_symint.
Support dynamism on "transpose" operation. (#4606)
* added failing test_t_copy test
* made transpose work
* disable the backward pass.
* fix linter
Add dynamic shape test for unsqueeze_copy (#4608)
* added failing test
* test correctness.
Support dynamism on view_copy_symint (#4629)
* Disable failing XLA tests
* Re-enable dynamo tests (#4454)
* Turn on keep going
* Re-enable more dynamic shape tests (#4558)
* fix size node ne op kind to be size_ne
* reenable test_nonzero_cast
* fix linter
* disable new failing test after rebase temporarily.
* fix linter
* [Functionalization] Fix ScatterReduce (#4576)
Summary:
ScatterReduce::reduce_ uses a unsafe type c10::string_view to store the
string. Replace it with std::string and re-enable all previous failing tests.
Test Plan:
TPU_LIBRARY_PATH=/home/ptxla/.local/lib/python3.8/site-packages/libtpu/libtpu.so PJRT_DEVICE=CPU test/cpp/build/test_ptxla --gtest_filter=AtenXlaTensorTest.TestScatterReduce*
* Revert "Turn on keep going"
This reverts commit 224bdf3d2c166e4e5a343d290f1b90b0d4308096.
* Turn on keep going
* Revert "Turn on keep going"
This reverts commit 224bdf3d2c166e4e5a343d290f1b90b0d4308096.
* Turn on keep going
* Revert "Turn on keep going"
This reverts commit 224bdf3d2c166e4e5a343d290f1b90b0d4308096.
* code compiles
* wrote new test TestDynamicShapes.test_sizeMod and it succeeds.
* the first view_copy_symint test test_view_copy_symint_with_dyn_input_shape passed
* added two more tests
* run linter
* added a new failing test with negative shape
* fix all tests
* clean up
* clean up
* fix a small test failure.
* fix pr comments
* updated the tests for dynamic input and static input shape
* not support dynamic input and static input shape.
* fix linter
* fix pr comments
* fix build error
* resolve a rebase conflict
* run linter
* add one more test
* add one more test and it passes.
* fix linter
* remove experiemental test.
---------
Co-authored-by: Wonjoo Lee <wonjoo@google.com>
Co-authored-by: Jiewen Tan <jwtan@google.com>
fix a merge error
fix other merge error during rebase.
* Track type of SymNode for XLASymNodeImpl::mod (#4732)
