DeepSpeed
48c5220b - Refactor residual add kernels (#2333)

Commit

2 years ago

Refactor residual add kernels (#2333) Co-authored-by: Ammar Ahmad Awan <ammar.awan@microsoft.com>

References

#2333 - Refactor residual add kernels

Author

arashb

arashb

Parents

Files3

csrc/transformer/inference/csrc
- pt_binding.cpp
deepspeed/ops/transformer/inference
- transformer_inference.py
tests/unit/ops/transformer/inference
- test_residual_add.py

Loading