Code

[Pallas] Make gmm functional#7117

Code

Merged

alanwaketan merged 10 commits intomasterpytorch/xla:masterfrom alanwaketan/gmm3pytorch/xla:alanwaketan/gmm3Copy head branch name to clipboardMay 26, 2024

+36-33

Commits

Commits on May 25, 2024

• use repeat_with_fixed_output_size

  alanwaketan

  committed
  453017d

  Copy full SHA for 453017d

• use _histogram

  alanwaketan

  committed
  615ace3

  Copy full SHA for 615ace3

• Make _make_group_metadata work for xla

  alanwaketan

  committed
  c754762

  Copy full SHA for c754762

• Improve test

  alanwaketan

  committed
  86925d7

  Copy full SHA for 86925d7

• make gmm functional on xla

  alanwaketan

  committed
  85e77c4

  Copy full SHA for 85e77c4

• Fix tests

  alanwaketan

  committed
  ef59145

  Copy full SHA for ef59145

• Fix linters

  alanwaketan

  committed
  7964f44

  Copy full SHA for 7964f44

• Fix a comment

  alanwaketan

  committed
  7e61cc3

  Copy full SHA for 7e61cc3

• Your commit message

  alanwaketan

  committed
  d41c4f4

  Copy full SHA for d41c4f4

• Fix test

  alanwaketan

  committed
  e5dde4a

  Copy full SHA for e5dde4a