oneflow icon indicating copy to clipboard operation
oneflow copied to clipboard

OneFlow is a deep learning framework designed to be user-friendly, scalable and efficient.

Results 591 oneflow issues
Sort by recently updated
recently updated
newest added

因为 NPU GPT2 测试场景下,layer_norm 都是 affine 的,为减少 CANN 调用,对 layer_norm_grad 进行重构。在一个 kernel 内完成 gamma,beta,dx 的梯度计算。