xla
916b1c4b - Support dynamism on view_copy_symint (#4629)

Commit
2 years ago
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>
Author
Committer
Parents
Loading