DeepSpeed
c5814647 - Openfold fix (#4368)

Commit
1 year ago
Openfold fix (#4368) * update * format --------- Co-authored-by: Conglong Li <conglong.li@gmail.com>
Author
Parents
  • csrc/deepspeed4science/evoformer_attn
    • File
      attention.cu
    • File
      attention_back.cu
    • File
      gemm_kernel_utils.h
    • transform
      • File
        bias_broadcast.h
  • op_builder
    • File
      evoformer_attn.py
  • tests/benchmarks
    • File
      DS4Sci_EvoformerAttention_bench.py