Code

triton softmax support multi-batch#152

Code

Open

Gy-Lu wants to merge 5 commits intomainhpcaitech/FastFold:mainfrom softmax_batchhpcaitech/FastFold:softmax_batchCopy head branch name to clipboard

+88-16

Commits

Commits on Feb 16, 2023

• modify fused softmax for batch dim

  Gy-Lu

  committed
  144d25f

  Copy full SHA for 144d25f

• polish

  Gy-Lu

  committed
  81b151a

  Copy full SHA for 81b151a

• polish

  Gy-Lu

  committed
  161d636

  Copy full SHA for 161d636

Commits on Feb 20, 2023

• support batch for triton softmax and add unit test

  Gy-Lu

  committed
  0884b00

  Copy full SHA for 0884b00

• remove redundent demension in test_softmax

  Gy-Lu

  committed
  c0876e3

  Copy full SHA for c0876e3