TransformerEngine : Add test with FSDP (and updates to ddp_wrapper
in test_ddp.py)
#118
Job | Run time |
---|---|
13s | |
13s |
ddp_wrapper
in test_ddp.py)
#118
Job | Run time |
---|---|
13s | |
13s |