Skip to content

Commit

Permalink
Add pax patches for CI testing (#1146)
Browse files Browse the repository at this point in the history
Signed-off-by: ashors1 <[email protected]>
  • Loading branch information
ashors1 authored Nov 14, 2024
1 parent cc2922c commit 59ec85c
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions .github/container/manifest.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,9 @@ paxml:
patches:
pull/46/head: file://patches/paxml/PR-46.patch # adds Transformer Engine support
pull/93/head: file://patches/paxml/PR-93.patch # exposes TE DPA argument in Grok to make it configurable from the command line
mirror/patch/validation-configs: file://patches/paxml/mirror-patch-validation-configs.patch # adds additional configs for testing
mirror/patch/log-avg-loss: file://patches/paxml/mirror-patch-log-avg-loss.patch # logs average loss over last `n` steps
mirror/patch/status-logging: file://patches/paxml/mirror-patch-status-logging.patch # sets log level of task status lines to warning
praxis:
url: https://github.com/google/praxis.git
mirror_url: https://github.com/nvjax-svc-0/praxis.git
Expand Down

0 comments on commit 59ec85c

Please sign in to comment.