Code

run sdpa with dtensor#163

Code

Closed

tianyu-l wants to merge 2 commits intogh/tianyu-l/5/basepytorch/torchtitan:gh/tianyu-l/5/basefrom gh/tianyu-l/5/headpytorch/torchtitan:gh/tianyu-l/5/headCopy head branch name to clipboard

+9-8

Commits

Commits on Mar 26, 2024

• run sdpa with dtensor

  Show description for 19f0912
  tianyu-l

  committed
  19f0912

  Copy full SHA for 19f0912

• Update on "run sdpa with dtensor"

  Show description for 32d90b5
  tianyu-l

  committed
  32d90b5

  Copy full SHA for 32d90b5