Uh oh!
There was an error while loading. Please reload this page.
[JAX] Cherry-pick #785 and #780 - #800
Merged
Merged
Conversation
…ax.jit (NVIDIA#785) * fixed static argnums for jax.jit in single gpu encoder test, changed warning filtering for pytest Signed-off-by: Alp Dener <adener@nvidia.com> * propagating the fix to the JAX mnist example Signed-off-by: Alp Dener <adener@nvidia.com> * fixed missing space ibetween flags i QAA scripts Signed-off-by: Alp Dener <adener@nvidia.com> * added TE warnings into the ignore list Signed-off-by: Alp Dener <adener@nvidia.com> --------- Signed-off-by: Alp Dener <adener@nvidia.com> Signed-off-by: Tim Moon <tmoon@nvidia.com>
…#780) * Allow multi-dims for dgamma and dbeta in LN descriptor. Signed-off-by: Ming Huang <mingh@nvidia.com> * Fix the jit error in examples/jax Signed-off-by: Ming Huang <mingh@nvidia.com> --------- Signed-off-by: Ming Huang <mingh@nvidia.com> Signed-off-by: Tim Moon <tmoon@nvidia.com>
timmoon10force-pushed
the
pr780-cherry-pick
branch
from
April 22, 2024 23:50
82c0141 to
e9ebc9bComparetimmoon10
commented
Apr 22, 2024
MemberAuthor
/te-ci jax |
ksivaman pushed a commit
that referenced
this pull request
Apr 24, 2024
* [JAX] Fixing CI failure due to incorrect use of `static_argnums` in jax.jit (#785) * fixed static argnums for jax.jit in single gpu encoder test, changed warning filtering for pytest Signed-off-by: Alp Dener <adener@nvidia.com> * propagating the fix to the JAX mnist example Signed-off-by: Alp Dener <adener@nvidia.com> * fixed missing space ibetween flags i QAA scripts Signed-off-by: Alp Dener <adener@nvidia.com> * added TE warnings into the ignore list Signed-off-by: Alp Dener <adener@nvidia.com> --------- Signed-off-by: Alp Dener <adener@nvidia.com> Signed-off-by: Tim Moon <tmoon@nvidia.com> * [JAX] Allow multi-dims for dgamma and dbeta in LN descriptor. (#780) * Allow multi-dims for dgamma and dbeta in LN descriptor. Signed-off-by: Ming Huang <mingh@nvidia.com> * Fix the jit error in examples/jax Signed-off-by: Ming Huang <mingh@nvidia.com> --------- Signed-off-by: Ming Huang <mingh@nvidia.com> Signed-off-by: Tim Moon <tmoon@nvidia.com> --------- Signed-off-by: Alp Dener <adener@nvidia.com> Signed-off-by: Tim Moon <tmoon@nvidia.com> Signed-off-by: Ming Huang <mingh@nvidia.com> Co-authored-by: Alp Dener <adener@nvidia.com> Co-authored-by: Ming-Xu Huang <mingh@nvidia.com>
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for freeto join this conversation on GitHub.
Already have an account?
Sign in to comment
4 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
See #785 and #780