Code

feat: add Ops.batch#535

Code

Draft

avik-pal wants to merge 13 commits intomainEnzymeAD/Reactant.jl:mainfrom ap/batchEnzymeAD/Reactant.jl:ap/batchCopy head branch name to clipboard

+477-256

Commits

Commits on Jan 22, 2025

• feat: implement Ops.batch

  avik-pal

  committed
  6322337

  Copy full SHA for 6322337

• refactor: cleanup elem_apply

  avik-pal

  committed
  11fdc36

  Copy full SHA for 11fdc36

• refactor: move elem_apply to ops and generalize batching in tracing

  avik-pal

  committed
  e5113bb

  Copy full SHA for e5113bb

• feat: support arbitrary structures for batching

  avik-pal

  committed
  c52022d

  Copy full SHA for c52022d

• feat: attempt a tracer impl for rewriting the args

  avik-pal

  committed
  fc03c4e

  Copy full SHA for fc03c4e

• feat: support arbitrary structures for batching

  avik-pal

  committed
  c509de5

  Copy full SHA for c509de5

• fix: tests used older defn

  avik-pal

  committed
  0862045

  Copy full SHA for 0862045

• fix: update traced_type

  avik-pal

  committed
  bcc4d59

  Copy full SHA for bcc4d59

• test: update call

  avik-pal

  committed
  6f776a0

  Copy full SHA for 6f776a0

• fix: tracing

  avik-pal

  committed
  0ad6330

  Copy full SHA for 0ad6330

• docs: setup batching tutorial

  avik-pal

  committed
  dd22219

  Copy full SHA for dd22219

• fix: call with reactant

  avik-pal

  committed
  f29babc

  Copy full SHA for f29babc

• fix: convert scalars to arrays in batcharray mode

  avik-pal

  committed
  2241e54

  Copy full SHA for 2241e54