Code

develop eager AMP#9088

Code

Merged

hjchen2 merged 26 commits intomasterOneflow-Inc/oneflow:masterfrom dev_eager_ampOneflow-Inc/oneflow:dev_eager_ampCopy head branch name to clipboardSep 18, 2022

Commits

Commits on Sep 13, 2022

• implement eager AMP

  hjchen2

  committed
  e443995

  Copy full SHA for e443995

• skip autocast for inplace and implement make autocast meta

  hjchen2

  committed
  cd45738

  Copy full SHA for cd45738

• fix

  hjchen2

  committed
  6006fef

  Copy full SHA for 6006fef

• Merge branch 'master' of https://github.com/Oneflow-Inc/oneflow into dev_eager_amp

  hjchen2

  committed
  fbf6992

  Copy full SHA for fbf6992

Commits on Sep 14, 2022

• rm unused code

  hjchen2

  committed
  dc5e2bc

  Copy full SHA for dc5e2bc

• autocast python api

  hjchen2

  committed
  981e266

  Copy full SHA for 981e266

• fix

  hjchen2

  committed
  7df529d

  Copy full SHA for 7df529d

• fix

  hjchen2

  committed
  0a31e5a

  Copy full SHA for 0a31e5a

Commits on Sep 15, 2022

• refine

  hjchen2

  committed
  5f10754

  Copy full SHA for 5f10754

• skip autocast if any input is float32 for gray or clear list

  hjchen2

  committed
  9be12ef

  Copy full SHA for 9be12ef

• refine

  hjchen2

  committed
  f9d420f

  Copy full SHA for f9d420f

• fix dead loop

  hjchen2

  committed
  a52cc5b

  Copy full SHA for a52cc5b

• add autocast unittest

  hjchen2

  committed
  2f349d8

  Copy full SHA for 2f349d8

• Merge branch 'master' into dev_eager_amp

  hjchen2

  authored
  9314382

  Copy full SHA for 9314382

• fix static analysis, refine as comments

  hjchen2

  committed
  301db7a

  Copy full SHA for 301db7a

• Merge branch 'dev_eager_amp' of https://github.com/Oneflow-Inc/oneflow into dev_eager_amp

  hjchen2

  committed
  d07fe15

  Copy full SHA for d07fe15

• Merge branch 'master' into dev_eager_amp

  hjchen2

  authored
  e8f84bd

  Copy full SHA for e8f84bd

• Merge branch 'master' into dev_eager_amp

  hjchen2

  authored
  0235c3e

  Copy full SHA for 0235c3e

• Merge branch 'master' into dev_eager_amp

  hjchen2

  authored
  57af7c7

  Copy full SHA for 57af7c7

Commits on Sep 16, 2022

• fix CUDA_VERSION >= 10000

  hjchen2

  authored
  6e4bd97

  Copy full SHA for 6e4bd97

• skip cuda autocast test

  hjchen2

  committed
  f5fdd35

  Copy full SHA for f5fdd35

• Merge branch 'dev_eager_amp' of https://github.com/Oneflow-Inc/oneflow into dev_eager_amp

  hjchen2

  committed
  8772cb2

  Copy full SHA for 8772cb2

• Merge branch 'master' into dev_eager_amp

  hjchen2

  authored
  230b0f6

  Copy full SHA for 230b0f6

• Merge branch 'master' into dev_eager_amp

  hjchen2

  authored
  565ab2a

  Copy full SHA for 565ab2a

• Merge branch 'master' into dev_eager_amp

  hjchen2

  authored
  b0c838f

  Copy full SHA for b0c838f

Commits on Sep 17, 2022

• Merge branch 'master' into dev_eager_amp

  hjchen2

  authored
  c0e190b

  Copy full SHA for c0e190b