AMD support #1430
jeffra
merged
41 commits into
master
from
staging-amd-v4
[squash] staging amd v3 (#1427)
fa52d9d3
formatting
700282b1
fix import error
aeeb444c
fix missing get_default_compute_capabilities
8124415d
set pip path
d9ce0901
install libaio-dev
9076a3d6
tjruwase
commented
on 2021-10-05
tjruwase
approved these changes
on 2021-10-05
Merge branch 'master' into staging-amd-v4
85b6ea43
stas00
approved these changes
on 2021-10-06
jeffwilcox2
approved these changes
on 2021-10-06
jeffra45
approved these changes
on 2021-10-06
bump DSE
a0c16c6c
Merge branch 'master' into staging-amd-v4
873cd511
Merge branch 'master' into staging-amd-v4
29d909a0
Merge branch 'master' into staging-amd-v4
1f901a2b
Merge branch 'master' into staging-amd-v4
0af7b1a3
Merge branch 'master' into staging-amd-v4
c171e2d7
Merge branch 'master' into staging-amd-v4
71086af1
Merge branch 'master' into staging-amd-v4
c209c05e
add pytest colors
e32e0fbb
fix LF issues
e33141e8
build autograd op and install autotuning extras
f12ddddc
Merge branch 'master' into staging-amd-v4
64c6df75
rocm workaround
0e691798
elevate perms for rocm patch
90a1de89
use rocm compatible cuda_home for cpu_adagrad
1359e6da
fix another cuda_home issue
0ee4ca43
Merge branch 'master' into staging-amd-v4
cb770434
add rocm support
4728bcb8
Merge branch 'master' into staging-amd-v4
1e57a0d3
remove extra set_train_batch_size in engine and fix submodule
d73d7df3
Added changes from https://github.com/ROCmSoftwarePlatform/DeepSpeed/…
475b41a0
skip 1bit tests if rocm > 4, cupy is not yet available for rocm 5
5e68a280
Merge branch 'master' into staging-amd-v4
8b00f9b1
spelling fixes
bdf693e4
workaround rocm-smi error
18f5e6de
fix cupy bug
9edfccae
use JIT install
d4809720
install into user writable path due outstanding to amd jit issue
71e5735f
fix to build cpu[adam|adagrad] on rocm
0679960b
sequential amd unit tests for now
4d63d7e8
remove extra -O3 in cpu-adam
38823042
fix weight init bug in zero test
acb5343f
fix ds_report error in CI, node having path issues
bc99f08e
Merge branch 'master' into staging-amd-v4
5ff469c7
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub