ds_transformer_cuda.h
Commit
4 years ago
Author
Reza Yazdani
Parents
- Files7
- csrc
- includes
- transformer
ds_transformer_cuda.cpp
- deepspeed
__init__.py- pt
- deepspeed_cuda.py
- tests/unit
- modelingpreln.py
- test_cuda_backward.py
- test_cuda_forward.py