fixed bugs in softmax kernel and update unittest for softmax

hpcaitech / FastFold

Optimizing AlphaFold Training and Inference on GPU Clusters

Apache License 2.0

559 stars 86 forks source link

Closed Shenggan closed 1 year ago

Shenggan commented 1 year ago

Gy-Lu commented 1 year ago

I have tested on real data, works well.