DeepSpeed/csrc/deepspeed4science/evoformer_attn/epilogue
..
epilogue_grad_bias.h
epilogue_pipelined.h
epilogue_rescale_output.h
epilogue_thread_apply_logsumexp.h