Code

Add FMHA PAXML test #830

Code

Open

hmonishN wants to merge 21 commits intomainNVIDIA/JAX-Toolbox:mainfrom hmonish/add_fmha_paxml_testNVIDIA/JAX-Toolbox:hmonish/add_fmha_paxml_testCopy head branch name to clipboard

+73-20

Commits

Commits on May 15, 2024

• add fmha related changes

  hmonishN

  authored
  a9a828b

  Copy full SHA for a9a828b

Commits on May 16, 2024

• Update _test_upstream_pax.yaml

  hmonishN

  authored
  69f9db9

  Copy full SHA for 69f9db9

• Update _sandbox.yaml

  hmonishN

  authored
  35565ce

  Copy full SHA for 35565ce

• Update _test_upstream_pax.yaml

  hmonishN

  authored
  42566a6

  Copy full SHA for 42566a6

• Update _sandbox.yaml

  hmonishN

  authored
  495ae32

  Copy full SHA for 495ae32

• Update test-pax.sh

  hmonishN

  authored
  23a19d8

  Copy full SHA for 23a19d8

• Update _sandbox.yaml

  hmonishN

  authored
  4be1f40

  Copy full SHA for 4be1f40

• Merge branch 'main' into hmonish/add_fmha_paxml_test

  hmonishN

  authored
  cb9d7d5

  Copy full SHA for cb9d7d5

• removing hlo dir for llama test.

  hmonishN

  authored
  f44cdef

  Copy full SHA for f44cdef

Commits on May 20, 2024

• Update _test_upstream_pax.yaml

  Show description for da69dbd
  hmonishN

  authored
  da69dbd

  Copy full SHA for da69dbd

• Update _test_upstream_pax.yaml

  hmonishN

  authored
  a6622c8

  Copy full SHA for a6622c8

• Update test-pax.sh

  Show description for b67229b
  hmonishN

  authored
  b67229b

  Copy full SHA for b67229b

• Update test-pax.sh

  Show description for f7618bf
  hmonishN

  authored
  f7618bf

  Copy full SHA for f7618bf

• Merge branch 'main' into hmonish/add_fmha_paxml_test

  hmonishN

  authored
  ab22fcc

  Copy full SHA for ab22fcc

Commits on Jun 10, 2024

• Update test-pax.sh

  Show description for dbb999d
  hmonishN

  authored
  dbb999d

  Copy full SHA for dbb999d

• Merge branch 'main' into hmonish/add_fmha_paxml_test

  hmonishN

  authored
  e0f4b4e

  Copy full SHA for e0f4b4e

Commits on Jun 11, 2024

• Merge branch 'main' into hmonish/add_fmha_paxml_test

  hmonishN

  authored
  1898b32

  Copy full SHA for 1898b32

• Update _test_pax_rosetta.yaml

  hmonishN

  authored
  c1ff8ae

  Copy full SHA for c1ff8ae

• Update _test_upstream_pax.yaml

  hmonishN

  authored
  0ef811a

  Copy full SHA for 0ef811a

• Update test-pax.sh

  hmonishN

  authored
  ca6e2e9

  Copy full SHA for ca6e2e9

• Update _test_upstream_pax.yaml

  hmonishN

  authored
  7084812

  Copy full SHA for 7084812