microsoft/DeepSpeed

Pull Requests Commits

tear down torch dist pg when test completes

jeffra committed 4 years ago

3beb9fd9

turn back on test_configurable_parallel

jeffra committed 4 years ago

1324589e

skip moe ckpt test if old torch

jeffra committed 4 years ago

08695e29

Merge branch 'master' into reyazda/test-sparse-v2

jeffra committed 4 years ago

Verified 80011245

fix: support three digit layer numbers (#1377)

adammoody committed 4 years ago

Verified 4ad8019c

add sparse-attn skip if not compatible

jeffra committed 4 years ago

7733f00d

fix megatron regression, add moe unit test, fix moe ckpt comparison

jeffra committed 4 years ago

7d9c5fb1

Merge branch 'reyazda/test-sparse-v2' of github.com:microsoft/DeepSpeed into reyazda/test-sparse-v2

Reza Yazdani committed 4 years ago

bd13efde

resolve the TK with correct setting based on dtype and block size

Reza Yazdani committed 4 years ago

940abc6c

fixes random connection reset test failures for some unit tests

jeffra committed 4 years ago

d1615ff0

jeffra committed 4 years ago

0d28040f

skip test_configurable_parallel for now until fixed

jeffra committed 4 years ago

35245fef

jeffra committed 4 years ago

45a498d0

[zero_to_fp32] adapt to 4-bytes alignment in z2 (#1372)

stas00 committed 4 years ago

Verified 30537e71

rename test_path so pytest doesn't think its a test

jeffra committed 4 years ago

61bf986e

fix relative import

jeffra committed 4 years ago

49514bd3

Merge branch 'master' into reyazda/test-sparse-v2

jeffra committed 4 years ago

Verified 99634357

refactor sparse attn imports

jeffra committed 4 years ago

318ae956

[zero Init] fix regression (#1373)

stas00 committed 4 years ago

Verified cf22a691

jeffra committed 4 years ago

626a51e4

factor out relative path for unit test files

jeffra committed 4 years ago

741f8b55

use relative paths for megatron jsons

jeffra committed 4 years ago

54752d0e

turn back on tests

jeffra committed 4 years ago

e7028669

add torch18 tests and fix sparse-attn checks

jeffra committed 4 years ago

9e2dee4b

update with assert and some fixes

jeffra committed 4 years ago

be55c85c

fixing the sparse-APIs to use the latest triton version

Reza Yazdani committed 4 years ago

80dbe2f7

let the sparse tests run

Reza Yazdani committed 4 years ago

18a8dffe

Introduce a device rank when setting device (#1370)

Sean Naren committed 4 years ago

Verified 90398a70

jeffra committed 4 years ago

a708b187

Update setup.py (#1361)

anukaal committed 4 years ago

Verified 8e577c92

Older