DeepSpeed

81b744ef - hip cooperative groups functionality for coalesced_group in fused_lamb_cuda_kernel.cu is not implemented yet