DeepSpeed
4017a7c9 - fix the workspace allocation for the transformer kernel

Commit

4 years ago

fix the workspace allocation for the transformer kernel

References

#1404 - Transformer/fix layer norm

#1397 - Fix the workspace allocation for the transformer kernel

Author

Reza Yazdani

Parents

Loading