Code

Sparse tensors#1998

Code

Draft

McArthur-Alford wants to merge 40 commits intotracel-ai:maintracel-ai/burn:mainfrom McArthur-Alford:sparse-tensorMcArthur-Alford/burn:sparse-tensorCopy head branch name to clipboard

+3,126-117

Commits

Commits on Jul 11, 2024

• sparse

  McArthur-Alford

  committed
  60ee8e6

  Copy full SHA for 60ee8e6

• Fixed errors from moving sparse

  McArthur-Alford

  committed
  24daafd

  Copy full SHA for 24daafd

• some better imports and fixes

  McArthur-Alford

  committed
  790ed5f

  Copy full SHA for 790ed5f

• Full sparse backend trait, lots unfinished

  McArthur-Alford

  committed
  6371a0a

  Copy full SHA for 6371a0a

• sparse_reshape op

  McArthur-Alford

  committed
  6f54864

  Copy full SHA for 6f54864

Commits on Jul 14, 2024

• permute and transpose working

  McArthur-Alford

  committed
  741d6cc

  Copy full SHA for 741d6cc

• swap dims

  McArthur-Alford

  committed
  08143fb

  Copy full SHA for 08143fb

• sparse flip

  McArthur-Alford

  committed
  19de621

  Copy full SHA for 19de621

• any, all, any_dim, all_dim

  McArthur-Alford

  committed
  c56c25d

  Copy full SHA for c56c25d

• repeat

  McArthur-Alford

  committed
  e75181d

  Copy full SHA for e75181d

Commits on Jul 16, 2024

• coalesce, somewhat broken

  McArthur-Alford

  committed
  cfe706b

  Copy full SHA for cfe706b

Commits on Jul 17, 2024

• fixed coalesce

  McArthur-Alford

  committed
  57f6dbb

  Copy full SHA for 57f6dbb

• fixed slice

  McArthur-Alford

  committed
  9a80208

  Copy full SHA for 9a80208

• sparse density

  McArthur-Alford

  committed
  2e0abd2

  Copy full SHA for 2e0abd2

• numeric for sparse tensors, and add

  McArthur-Alford

  committed
  60d8f67

  Copy full SHA for 60d8f67

Commits on Jul 20, 2024

• add, sub, mul, div and some refactors

  McArthur-Alford

  committed
  1d9f856

  Copy full SHA for 1d9f856

• slice_assign

  McArthur-Alford

  committed
  4782a3b

  Copy full SHA for 4782a3b

Commits on Jul 21, 2024

• sddmm + more numerics (sign, abs, etc)

  McArthur-Alford

  committed
  f898d79

  Copy full SHA for f898d79

Commits on Jul 28, 2024

• made unimplemented functions panic

  McArthur-Alford

  committed
  80ab5d8

  Copy full SHA for 80ab5d8

Commits on Aug 11, 2024

• style fixes

  McArthur-Alford

  committed
  f6c0ff8

  Copy full SHA for f6c0ff8

• Merge branch 'main' of github.com:tracel-ai/burn into sparse-tensor

  McArthur-Alford

  committed
  a67cb0a

  Copy full SHA for a67cb0a

• Refactor of tensor API in progress

  McArthur-Alford

  committed
  b63ea7a

  Copy full SHA for b63ea7a

Commits on Aug 14, 2024

• New sparse tensor API, seems really good

  McArthur-Alford

  committed
  1d0d366

  Copy full SHA for 1d0d366

• Changing up primitives for blanket impl

  McArthur-Alford

  committed
  b98ecfc

  Copy full SHA for b98ecfc

• Seemingly everything but tensorchecks working

  McArthur-Alford

  committed
  6ef4b4d

  Copy full SHA for 6ef4b4d

Commits on Aug 19, 2024

• Reintroduced the COO decorator to burn-sparse

  McArthur-Alford

  committed
  5d53f7a

  Copy full SHA for 5d53f7a

• transferred accross most basic ops for float tensor

  McArthur-Alford

  committed
  6ebe15b

  Copy full SHA for 6ebe15b

• most functions transferred

  McArthur-Alford

  committed
  937480e

  Copy full SHA for 937480e

Commits on Aug 20, 2024

• Added use to mod

  McArthur-Alford

  committed
  bac85ac

  Copy full SHA for bac85ac

Commits on Aug 21, 2024

• Some more functions, a little broken

  McArthur-Alford

  committed
  1dac1e8

  Copy full SHA for 1dac1e8

Commits on Aug 22, 2024

• A huge overhaul, much nicer types and much less confusing, achieves the same result

  McArthur-Alford

  committed
  6150e8a

  Copy full SHA for 6150e8a

• Cleanup of types

  McArthur-Alford

  committed
  40d2afd

  Copy full SHA for 40d2afd

Commits on Aug 24, 2024

• BasicSparseOps & into/from sparse

  McArthur-Alford

  committed
  1c06aab

  Copy full SHA for 1c06aab

Commits on Aug 25, 2024

• Big cleanup of burn-sparse

  McArthur-Alford

  committed
  4535e37

  Copy full SHA for 4535e37

• Removed old

  McArthur-Alford

  committed
  ae8ab68

  Copy full SHA for ae8ab68

• Removed unsupported sparse ops

  McArthur-Alford

  committed
  7b90252

  Copy full SHA for 7b90252

Commits on Aug 30, 2024

• Coordinates OP, plus basicsparse for float/int

  McArthur-Alford

  committed
  e00de1e

  Copy full SHA for e00de1e

• Removed unsupported ops

  McArthur-Alford

  committed
  d8603f3

  Copy full SHA for d8603f3

• Merge branch 'sparse-tensor' of github.com:McArthur-Alford/burn into sparse-tensor

  McArthur-Alford

  committed
  ed46fd2

  Copy full SHA for ed46fd2

Commits on Oct 2, 2024

• values

  McArthur-Alford

  committed
  cedd197

  Copy full SHA for cedd197