Code

[Metal] Scaled masked softmax#1602

Code

Merged

kali merged 6 commits intomainsonos/tract:mainfrom task/scaled-masked-softmaxsonos/tract:task/scaled-masked-softmaxCopy head branch name to clipboardDec 11, 2024

+511-5

Commits

Commits on Dec 10, 2024

• Metal Implementatio of Scaled Masked Softmax

  hubertdelajonquieresonos

  committed
  8ed325a

• WIP

  hubertdelajonquieresonos

  committed
  b135455

• Scaled Masked Softmax integration WIP

  hubertdelajonquieresonos

  committed
  dab1e94

Commits on Dec 11, 2024

• More test

  hubertdelajonquieresonos

  committed
  c22bb42

• Fix typo in comment

  hubertdelajonquieresonos

  committed
  1c85939

• Improve comments

  hubertdelajonquieresonos

  committed
  48e3947