Code

Quantized SDPA#1515

Code

Draft

barronalex wants to merge 5 commits intomainml-explore/mlx:mainfrom q-sdpaml-explore/mlx:q-sdpaCopy head branch name to clipboard

+747-108

Commits

Commits on Dec 6, 2024

• working qsdpa

  

  Alex Barron

  committed
  12a4d89

• add test

  

  Alex Barron

  committed
  3507c10

• add checks

  

  Alex Barron

  committed
  c89ddf6

• cpu fallback

  

  Alex Barron

  committed
  7697046

• fix test

  

  Alex Barron

  committed
  82a956c