DeepSpeed
3af2e4bb - simplify softmax kernel

Commit

2 years ago

simplify softmax kernel

References

mosm/softmax-longseq

Author

molly-smith

molly-smith

Parents

Files2

csrc/transformer/inference/csrc
- softmax.cu
op_builder
- builder.py