From 8f667e988dc69a9d3977b23ec8dd1eb26233d58e Mon Sep 17 00:00:00 2001 From: Shawn Wang Date: Tue, 8 Oct 2024 19:55:45 -0700 Subject: [PATCH] Add cuda profiler hook to paxml --- .github/container/manifest.yaml | 1 + 1 file changed, 1 insertion(+) diff --git a/.github/container/manifest.yaml b/.github/container/manifest.yaml index e9d30a3bc..17e454d89 100644 --- a/.github/container/manifest.yaml +++ b/.github/container/manifest.yaml @@ -40,6 +40,7 @@ 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 + pull/98/head: file://patches/paxml/PR-98.patch # Add cuda profiler hook function call to paxml praxis: url: https://github.com/google/praxis.git mirror_url: https://github.com/nvjax-svc-0/praxis.git