DeepSpeed
8263f200 - fix softmax BWD kernel

Commit

2 years ago

fix softmax BWD kernel

Author

Reza Yazdani

Parents

Files2

csrc/transformer
- softmax_kernels.cu
tests/unit
- test_softmax.py

Loading