DeepSpeed
644fea44 - fixing some bug in softmax kernel for batch_size>1

Commit

3 years ago

fixing some bug in softmax kernel for batch_size>1

References

#2083 - Add Inference support for running the BigScience-BLOOM Architecture

Author

Reza Yazdani

Parents

Loading