Code

(WIP) Batched autodiff#2181

Code

Draft

jumerckx wants to merge 12 commits intoEnzymeAD:mainEnzymeAD/Enzyme:mainfrom jumerckx:batchedjumerckx/Enzyme:batchedCopy head branch name to clipboard

+133-7

Commits

Commits on Dec 16, 2024

• add type conversions for width != 1.

  Show description for 8d3ef08
  jumerckx

  committed
  8d3ef08

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

  jumerckx

  committed
  5e647e5

• a test and formatting

  jumerckx

  committed
  f69f974

• use tensor splatop

  jumerckx

  committed
  623fccf

• remove stale enzyme-tblgen changes

  jumerckx

  committed
  f860e19

• do the simple batching in enzyme-tblgen

  jumerckx

  committed
  3a3bdf8

• include tensor in all AutoDiffOpInterfaceImpls

  jumerckx

  committed
  f1b2a6d

• add enzyme broadcastop

  jumerckx

  committed
  396a29b

• getShadowType for TensorTypeInterface

  jumerckx

  committed
  64489a5

• create broadcastop in enzyme-tblgen

  jumerckx

  committed
  0ccb37f

• Revert "include tensor in all AutoDiffOpInterfaceImpls"

  Show description for 123a985
  jumerckx

  committed
  123a985

• test

  jumerckx

  committed
  3d2911d