Code

Adding CPU training support to AxoNN#39

Code

Open

Avuxon wants to merge 28 commits intodevelopaxonn-ai/axonn:developfrom axonn-cpuaxonn-ai/axonn:axonn-cpuCopy head branch name to clipboard

+140-59

Commits

Commits on Feb 2, 2024

• Rebased cpu changes onto latest axonn

  

  Sathwik Yanamaddi

  committed
  e3bfa64

  Copy full SHA for e3bfa64

Commits on Feb 5, 2024

• Args should default to true, removed tensor_list operator for cpus

  

  Sathwik Yanamaddi

  committed
  472d355

  Copy full SHA for 472d355

Commits on Feb 6, 2024

• Made changes conditional for CPUs/GPUs

  

  Sathwik Yanamaddi

  committed
  9c237b1

  Copy full SHA for 9c237b1

Commits on Feb 8, 2024

• Gpu runs do not make Slurm calls

  

  Sathwik Yanamaddi

  committed
  676640d

  Copy full SHA for 676640d

Commits on Apr 12, 2024

• Revamped cpu flag with cpu hard set in comm.py

  Show description for e84c64d
  

  Sathwik Yanamaddi

  committed
  e84c64d

  Copy full SHA for e84c64d

• Set device before init

  Show description for ae74e70
  

  Sathwik Yanamaddi

  authored and

  Sathwik Yanamaddi

  committed
  ae74e70

  Copy full SHA for ae74e70

• Added initial pytest parameters for cpu tests

  

  Sathwik Yanamaddi

  authored and

  Sathwik Yanamaddi

  committed
  9514d65

  Copy full SHA for 9514d65

• removed mpi4py dependency (#63)

  Show description for 0a22f88
  

  2 peopleauthored and

  Sathwik Yanamaddi

  committed
  0a22f88

  Copy full SHA for 0a22f88

• adding parallelize context for opt (#65)

  

  jwendlan

  authored and

  Sathwik Yanamaddi

  committed
  52b0aef

  Copy full SHA for 52b0aef

• Removing the drop and gathers in depth tensor parallelism for the easy API (#66)

  

  siddharth9820

  authored and

  Sathwik Yanamaddi

  committed
  7728f1a

  Copy full SHA for 7728f1a

• change parallelize context to use AutoConfig (#67)

  

  siddharth9820

  authored and

  Sathwik Yanamaddi

  committed
  5087268

  Copy full SHA for 5087268

• Bugfix: Initialize grad_input, grad_weight to None (#68)

  Show description for 5faec5b
  

  adityaranjan

  authored and

  Sathwik Yanamaddi

  committed
  5faec5b

  Copy full SHA for 5faec5b

• Removed distributed communication class

  Show description for e831180
  

  Sathwik Yanamaddi

  committed
  e831180

  Copy full SHA for e831180

Commits on May 1, 2024

• Merge branch 'develop' into axonn-cpu

  siddharth9820

  authored
  0444e50

  Copy full SHA for 0444e50

• formatting

  siddharth9820

  committed
  70a5a37

  Copy full SHA for 70a5a37

• fix CI

  siddharth9820

  committed
  e185214

  Copy full SHA for e185214

• format

  siddharth9820

  committed
  e9f21ce

  Copy full SHA for e9f21ce

Commits on May 8, 2024

• Matched changes for intra_layer_conv tests

  

  Sathwik Yanamaddi

  committed
  f3a5dfc

  Copy full SHA for f3a5dfc

• Removed unused env variables and format

  

  Sathwik Yanamaddi

  committed
  88176e9

  Copy full SHA for 88176e9

• Merge branch 'develop' into axonn-cpu

  Avuxon

  authored
  c63e489

  Copy full SHA for c63e489

• Fixed CI to return for depth > 1

  

  Sathwik Yanamaddi

  committed
  f933edc

  Copy full SHA for f933edc

• Merge branch 'axonn-cpu' of https://github.com/axonn-ai/axonn into axonn-cpu

  

  Sathwik Yanamaddi

  committed
  20af63a

  Copy full SHA for 20af63a

• Added depth check to fw pass in fc test

  

  Sathwik Yanamaddi

  committed
  8249add

  Copy full SHA for 8249add

Commits on May 24, 2024

• Merge develop into this branch

  

  Sathwik Yanamaddi

  committed
  1a677ff

  Copy full SHA for 1a677ff

• Added xfail to convolution tests

  

  Sathwik Yanamaddi

  committed
  f39883f

  Copy full SHA for f39883f

Commits on Jun 10, 2024

• remove xfail

  siddharth9820

  committed
  d24fa4c

  Copy full SHA for d24fa4c

Commits on Jun 18, 2024

• Merge branch 'develop' into axonn-cpu

  siddharth9820

  authored
  ad779c3

  Copy full SHA for ad779c3

Commits on Jun 26, 2024

• Merge branch 'develop' into axonn-cpu

  siddharth9820

  authored
  1f9e54b

  Copy full SHA for 1f9e54b