DeepSpeed
Inference PP changes for neox
#1899
Merged

Inference PP changes for neox #1899

jeffra merged 30 commits into master from inference/pipechanges
jeffra
Synchronize the GPUs for the text-generation inference test
64a58c7c
change pp to run inference
d153b096
jeffra support ckpt conversion between pp<->seq
ec0edd55
jeffra remove conversion code, its now moved to client side
a145129b
add the inference-api changes to support the GPT-Neox inference at de…
1aa51481
fix the accuracy issue on the GPTNeoX inference
00d3be68
fixing parameter copy
525e9e7b
jeffra add inference global
7e965811
jeffra jeffra requested a review from samyam samyam 3 years ago
jeffra jeffra requested a review from tjruwase tjruwase 3 years ago
jeffra jeffra requested a review from ShadenSmith ShadenSmith 3 years ago
jeffra jeffra requested a review from conglongli conglongli 3 years ago
jeffra jeffra requested a review from awan-10 awan-10 3 years ago
jeffra jeffra requested a review from cli99 cli99 3 years ago
jeffra jeffra requested a review from eltonzheng eltonzheng 3 years ago
jeffra jeffra requested a review from minjiaz minjiaz 3 years ago
jeffra jeffra requested a review from RezaYazdaniAminabadi RezaYazdaniAminabadi 3 years ago
jeffra Merge branch 'master' into inference/pipechanges
69b611e3
jeffra
jeffra commented on 2022-04-19
jeffra
jeffra commented on 2022-04-19
jeffra
jeffra commented on 2022-04-19
jeffra
jeffra commented on 2022-04-19
jeffra
jeffra commented on 2022-04-19
jeffra
jeffra commented on 2022-04-19
jeffra
jeffra commented on 2022-04-19
jeffra
jeffra commented on 2022-04-19
jeffra jeffra changed the title Inference/pipechanges Inference PP changes for neox 3 years ago
jeffra clean-up
c271cc7c
jeffra cleanup
586c93fa
jeffra more clean-up
676023e0
jeffra Merge branch 'master' into inference/pipechanges
da45d1a7
jeffra more clean-up/docs
73afc65d
jeffra formatting
a31a93d4
fix minor issue with injection
cecde181
Merge branch 'inference/pipechanges' of github.com:microsoft/DeepSpee…
e3e0a9a8
refine the pipe module to get a map of the fwd_funcs to be able to in…
9e5e1ce0
jeffra fix DSE
afdcac42
jeffra add final_linear to state dict factory ckpt split
a16adb93
jeffra Merge branch 'master' into inference/pipechanges
7b3a75e0
jeffra fix for circular import issue
a82ae859
jeffra Merge branch 'inference/pipechanges' of github.com:microsoft/DeepSpee…
9eff79c6
jeffra comment out sanity check for now
3a3109c6
RezaYazdaniAminabadi Merge branch 'master' into inference/pipechanges
346186bb
fix layernorm-param issue & compilation on newer version of torch
3f490aec
change the default trainig_mp_size to 1
1891e300
jeffra Merge branch 'master' into inference/pipechanges
8bd754f4
jeffra
jeffra fix for curand on amd
5afbee96
jeffra fix syntax error
8627db77
RezaYazdaniAminabadi
RezaYazdaniAminabadi approved these changes on 2022-04-26
jeffra jeffra merged b4fcd98f into master 3 years ago
jeffra jeffra deleted the inference/pipechanges branch 3 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone