fix row_idx overflow in triton 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

when squence length extremely long, row_idx may overflow with int32. especially in extra_msa.