Code

Add images with flash attention 2#651

Code

Merged

dakinggg merged 26 commits intomainmosaicml/llm-foundry:mainfrom flash2-upstreammosaicml/llm-foundry:flash2-upstreamCopy head branch name to clipboardOct 10, 2023

+100-24

Commits

Commits on Oct 6, 2023

• wip

  dakinggg

  committed
  feaf129

Commits on Oct 7, 2023

• build images on prs that touch docker stuff

  dakinggg

  committed
  188f8aa

• precommit

  dakinggg

  committed
  6c6a80b

• add strings

  dakinggg

  committed
  9b26b23

• fix if statement

  dakinggg

  committed
  bde949f

• echo to env

  dakinggg

  committed
  2d041bc

• Merge branch 'main' into flash2-upstream

  dakinggg

  authored
  9cb4608

Commits on Oct 9, 2023

• add flash2 images

  dakinggg

  committed
  bebd26d

• precommit

  dakinggg

  committed
  969b3e7

• try again

  dakinggg

  committed
  554c2e3

• Merge branch 'main' into flash2-upstream

  dakinggg

  committed
  8dd16d1

• fix

  dakinggg

  committed
  1a82b74

• precommit

  dakinggg

  committed
  f65bfce

• fix version

  dakinggg

  committed
  fdec363

Commits on Oct 10, 2023

• gate attn impls

  dakinggg

  committed
  ba8dd76

• precommit

  dakinggg

  committed
  8144b6b

• yamllint

  dakinggg

  committed
  e4fd12a

• Merge branch 'main' into flash2-upstream

  dakinggg

  authored
  ca8d8d9

• Merge branch 'main' into flash2-upstream

  dakinggg

  authored
  dfee60e

• fix API and revert temp change

  dakinggg

  committed
  e4d0244

• remove comment

  dakinggg

  committed
  cbbbf1d

• update error message

  dakinggg

  committed
  64e2f03

• fix version typo

  dakinggg

  committed
  faa562d

• precommit

  dakinggg

  committed
  3f2f11a

• Merge branch 'main' into flash2-upstream

  dakinggg

  authored
  82eec57

• Merge branch 'main' into flash2-upstream

  dakinggg

  authored
  46613dd