Skip to content

Commit 8f667e9

Browse files
committed
Add cuda profiler hook to paxml
1 parent d1ff3c8 commit 8f667e9

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

.github/container/manifest.yaml

+1
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ paxml:
4040
patches:
4141
pull/46/head: file://patches/paxml/PR-46.patch # adds Transformer Engine support
4242
pull/93/head: file://patches/paxml/PR-93.patch # exposes TE DPA argument in Grok to make it configurable from the command line
43+
pull/98/head: file://patches/paxml/PR-98.patch # Add cuda profiler hook function call to paxml
4344
praxis:
4445
url: https://github.com/google/praxis.git
4546
mirror_url: https://github.com/nvjax-svc-0/praxis.git

0 commit comments

Comments
 (0)