Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

feat(jax): build nlist in the SavedModel & fix nopbc for StableHLO and SavedModel #4318

Merged
merged 32 commits into from
Nov 11, 2024
Merged

feat(jax): build nlist in the SavedModel & fix nopbc for StableHLO and SavedModel #4318

merged 32 commits into from
Nov 11, 2024

Commits on Nov 6, 2024

  1. checkpoint 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 6, 2024
    Configuration menu
    Copy the full SHA
    d924013 View commit details
    Browse the repository at this point in the history

  2. bugfix 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 6, 2024
    Configuration menu
    Copy the full SHA
    373ea65 View commit details
    Browse the repository at this point in the history

  3. bugfix 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 6, 2024
    Configuration menu
    Copy the full SHA
    933e4df View commit details
    Browse the repository at this point in the history

  4. nopbc 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 6, 2024
    Configuration menu
    Copy the full SHA
    94d2054 View commit details
    Browse the repository at this point in the history

  5. clean up default values 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz authored Nov 6, 2024
    Configuration menu
    Copy the full SHA
    84cb819 View commit details
    Browse the repository at this point in the history

Commits on Nov 7, 2024

  1. add tests 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 7, 2024
    Configuration menu
    Copy the full SHA
    bd27d4f View commit details
    Browse the repository at this point in the history

  2. skip the whole module 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 7, 2024
    Configuration menu
    Copy the full SHA
    a00aae8 View commit details
    Browse the repository at this point in the history

  3. fix tests on ci 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 7, 2024
    Configuration menu
    Copy the full SHA
    95c476b View commit details
    Browse the repository at this point in the history

  4. fix 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 7, 2024
    Configuration menu
    Copy the full SHA
    8bef185 View commit details
    Browse the repository at this point in the history

  5. fix skip testing. I am still confused why it doesn't work 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 7, 2024
    Configuration menu
    Copy the full SHA
    5e10621 View commit details
    Browse the repository at this point in the history

  6. try to resolve OOM issue 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 7, 2024
    Configuration menu
    Copy the full SHA
    fc7b6b7 View commit details
    Browse the repository at this point in the history

Commits on Nov 8, 2024

  1. Merge branch 'devel' into tf-call

    @njzjz
    njzjz authored Nov 8, 2024
    Configuration menu
    Copy the full SHA
    9f9c174 View commit details
    Browse the repository at this point in the history

  2. limit threads during tests 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 8, 2024
    Configuration menu
    Copy the full SHA
    31f9e87 View commit details
    Browse the repository at this point in the history

  3. set xla flags before any imports 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 8, 2024
    Configuration menu
    Copy the full SHA
    050c20c View commit details
    Browse the repository at this point in the history

  4. set NPROC 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 8, 2024
    Configuration menu
    Copy the full SHA
    19f798c View commit details
    Browse the repository at this point in the history

  5. I don;t understand why the tests fail randomly agter I add some tests... 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 8, 2024
    Configuration menu
    Copy the full SHA
    befc0c7 View commit details
    Browse the repository at this point in the history

  6. typo 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 8, 2024
    Configuration menu
    Copy the full SHA
    89e8371 View commit details
    Browse the repository at this point in the history

  7. release memory? 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 8, 2024
    Configuration menu
    Copy the full SHA
    8570079 View commit details
    Browse the repository at this point in the history

  8. --cov-append

    @njzjz
    njzjz authored Nov 8, 2024
    Configuration menu
    Copy the full SHA
    89041ed View commit details
    Browse the repository at this point in the history

Commits on Nov 9, 2024

  1. try scope module

    @njzjz
    njzjz authored Nov 9, 2024
    Configuration menu
    Copy the full SHA
    b9ff755 View commit details
    Browse the repository at this point in the history

  2. Revert "try scope module" 

    This reverts commit b9ff755.
    @njzjz
    njzjz committed Nov 9, 2024
    Configuration menu
    Copy the full SHA
    b1b8dd9 View commit details
    Browse the repository at this point in the history

  3. Revert "release memory?" 

    This reverts commit 8570079.
    @njzjz
    njzjz committed Nov 9, 2024
    Configuration menu
    Copy the full SHA
    48dca4b View commit details
    Browse the repository at this point in the history

  4. Revert "typo" 

    This reverts commit 89e8371.
    @njzjz
    njzjz committed Nov 9, 2024
    Configuration menu
    Copy the full SHA
    8d7a230 View commit details
    Browse the repository at this point in the history

  5. Revert "I don;t understand why the tests fail randomly agter I add so… 

    …me tests..."

This reverts commit befc0c7.
    @njzjz
    njzjz committed Nov 9, 2024
    Configuration menu
    Copy the full SHA
    5762a31 View commit details
    Browse the repository at this point in the history

  6. Revert "set NPROC" 

    This reverts commit 19f798c.
    @njzjz
    njzjz committed Nov 9, 2024
    Configuration menu
    Copy the full SHA
    ee831e0 View commit details
    Browse the repository at this point in the history

  7. Revert "set xla flags before any imports" 

    This reverts commit 050c20c.
    @njzjz
    njzjz committed Nov 9, 2024
    Configuration menu
    Copy the full SHA
    e6953c9 View commit details
    Browse the repository at this point in the history

  8. Revert "limit threads during tests" 

    This reverts commit 31f9e87.
    @njzjz
    njzjz committed Nov 9, 2024
    Configuration menu
    Copy the full SHA
    c365cb6 View commit details
    Browse the repository at this point in the history

  9. Revert "try to resolve OOM issue" 

    This reverts commit fc7b6b7.
    @njzjz
    njzjz committed Nov 9, 2024
    Configuration menu
    Copy the full SHA
    c94302d View commit details
    Browse the repository at this point in the history

  10. move the time-comsuming test out of main test 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 9, 2024
    Configuration menu
    Copy the full SHA
    b0a496c View commit details
    Browse the repository at this point in the history

  11. try 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 9, 2024
    Configuration menu
    Copy the full SHA
    4e42105 View commit details
    Browse the repository at this point in the history

  12. try to move jax2tf_tests to a different directory 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 9, 2024
    Configuration menu
    Copy the full SHA
    215efc6 View commit details
    Browse the repository at this point in the history

  13. clean up 

    Signed-off-by: Jinzhe Zeng <[email protected]>
    @njzjz
    njzjz committed Nov 9, 2024
    Configuration menu
    Copy the full SHA
    e4bac35 View commit details
    Browse the repository at this point in the history