Code

Batched autodiff#2181

Code

Merged

wsmoses merged 22 commits intoEnzymeAD:mainEnzymeAD/Enzyme:mainfrom jumerckx:batchedjumerckx/Enzyme:batchedCopy head branch name to clipboardDec 27, 2024

+239-11

Commits

Commits on Dec 16, 2024

• add type conversions for width != 1.

  Show description for 8d3ef08
  jumerckx

  committed
  8d3ef08

  Copy full SHA for 8d3ef08

• add code to tblgen generator, this eventually needs to be a single function call.

  jumerckx

  committed
  5e647e5

  Copy full SHA for 5e647e5

• a test and formatting

  jumerckx

  committed
  f69f974

  Copy full SHA for f69f974

• use tensor splatop

  jumerckx

  committed
  623fccf

  Copy full SHA for 623fccf

• remove stale enzyme-tblgen changes

  jumerckx

  committed
  f860e19

  Copy full SHA for f860e19

• do the simple batching in enzyme-tblgen

  jumerckx

  committed
  3a3bdf8

  Copy full SHA for 3a3bdf8

• include tensor in all AutoDiffOpInterfaceImpls

  jumerckx

  committed
  f1b2a6d

  Copy full SHA for f1b2a6d

• add enzyme broadcastop

  jumerckx

  committed
  396a29b

  Copy full SHA for 396a29b

• getShadowType for TensorTypeInterface

  jumerckx

  committed
  64489a5

  Copy full SHA for 64489a5

• create broadcastop in enzyme-tblgen

  jumerckx

  committed
  0ccb37f

  Copy full SHA for 0ccb37f

• Revert "include tensor in all AutoDiffOpInterfaceImpls"

  Show description for 123a985
  jumerckx

  committed
  123a985

  Copy full SHA for 123a985

• test

  jumerckx

  committed
  3d2911d

  Copy full SHA for 3d2911d

Commits on Dec 22, 2024

• DenseI64ArrayAttr for shape instead of scalar width

  

  Jules Merckx

  committed
  8a3f3c7

  Copy full SHA for 8a3f3c7

• llvm::SmallVector --> ArrayRef

  jumerckx

  committed
  aa4e598

  Copy full SHA for aa4e598

Commits on Dec 23, 2024

• formatting

  jumerckx

  committed
  fa190a4

  Copy full SHA for fa190a4

Commits on Dec 25, 2024

• use getShadowType in BroadcastOp builder

  Show description for 1802949
  

  jumerckx

  and

  wsmoses

  committed
  1802949

  Copy full SHA for 1802949

• unstructured control flow test

  jumerckx

  committed
  f9746ac

  Copy full SHA for f9746ac

• scf.for

  jumerckx

  committed
  2fb1412

  Copy full SHA for 2fb1412

• formatting

  jumerckx

  committed
  2c4f1fb

  Copy full SHA for 2c4f1fb

• support scf.if test

  jumerckx

  committed
  236b65d

  Copy full SHA for 236b65d

• formatting

  jumerckx

  committed
  9ceb922

  Copy full SHA for 9ceb922

• forgotten includes

  jumerckx

  committed
  50522a0

  Copy full SHA for 50522a0