Fix transpose op (#4695)
* impl pack in transpose util
* fix transpose sbp
* rm pack in transpose kernel
* fix bug
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/core/kernel/util/cuda_arithemetic_interface.cu 35 additions, 3 deletionsoneflow/core/kernel/util/cuda_arithemetic_interface.cu
- oneflow/user/kernels/transpose_kernel.cpp 3 additions, 22 deletionsoneflow/user/kernels/transpose_kernel.cpp
- oneflow/user/ops/transpose_ops.cpp 1 addition, 1 deletiononeflow/user/ops/transpose_ops.cpp
Please register or sign in to comment