DeepSpeed
DeepSpeed copied to clipboard
Add some improvements for pipeline module, engine and assertion into ds engine
There are some improvements.
- Improve error message of pipeline engine. see https://github.com/microsoft/DeepSpeed/pull/1438
- Move
has_bool_tensorstokwargsto make usable withdeepspeed.initializeand renamehastosend. see https://github.com/microsoft/DeepSpeed/pull/1399 - Add assertion to DS engine. see https://github.com/microsoft/DeepSpeed/issues/1034
- Add
@propertyto makemodule.mpuusable.- when we use
PipelineEngine, we use syntax likeengine.mpu. but when we usePipelineModulewe must usemodule.mpu(). this can be confusing front end users. - So, I added
@propertyto makemodule.mpuusable. We can use same syntax likesomething.mpu. - I also added
__call__()method toPipelineParallelGridto be compatible with exsting code
- when we use
cc. @tjruwase
Can one of the admins verify this patch?
Hi @hyunwoongko - apologies for taking so long to reply to this. Is this still a branch you'd like to see merged? If so, could you resolve merge conflicts? If not, we're closing and cleaning up old branches in the repo.