Restruct Elementwise kernel (#4130)
* init * first restruct version * remove useless code * add xpu_kernel class for elementwise ops * xpu_kernel class for elementwise finished * remove useless code * Add Binary template and elu backward * rename macro * fix params order * Add Ternary elemwise class * add simple ternary op to test. will be deleted after restructed * remove useless code * restruct elu * remove useless code * refine * restruct hardswish * restruct activation * remove useless code * rename activation op * remove elu useless code * remove hardswish useless code * fix file dir * restruct hardsigmoid * remove useless hardtanh code * fix double to float * restruct hardtanh * fix elu params * add space * refine * refine format * test variable args macro * refine header and macro * rename code file * remove code * rename launcher * rename macro * change header file name * restruct macro Co-authored-by:doombeaker <later@usopp.net> Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/core/framework/user_op_kernel_registry.h 0 additions, 1 deletiononeflow/core/framework/user_op_kernel_registry.h
- oneflow/user/kernels/activation_kernels.cpp 28 additions, 0 deletionsoneflow/user/kernels/activation_kernels.cpp
- oneflow/user/kernels/activation_kernels.cu 69 additions, 0 deletionsoneflow/user/kernels/activation_kernels.cu
- oneflow/user/kernels/activation_kernels.h 184 additions, 0 deletionsoneflow/user/kernels/activation_kernels.h
- oneflow/user/kernels/elementwise_xpu_kernel.cuh 41 additions, 0 deletionsoneflow/user/kernels/elementwise_xpu_kernel.cuh
- oneflow/user/kernels/elementwise_xpu_kernel.h 161 additions, 0 deletionsoneflow/user/kernels/elementwise_xpu_kernel.h
- oneflow/user/kernels/elu_kernel.cpp 0 additions, 90 deletionsoneflow/user/kernels/elu_kernel.cpp
- oneflow/user/kernels/elu_kernel.cu 0 additions, 127 deletionsoneflow/user/kernels/elu_kernel.cu
- oneflow/user/kernels/hardsigmoid_kernel.cpp 0 additions, 94 deletionsoneflow/user/kernels/hardsigmoid_kernel.cpp
- oneflow/user/kernels/hardsigmoid_kernel.cu 0 additions, 124 deletionsoneflow/user/kernels/hardsigmoid_kernel.cu
- oneflow/user/kernels/hardswish_kernel.cpp 0 additions, 99 deletionsoneflow/user/kernels/hardswish_kernel.cpp
- oneflow/user/kernels/hardswish_kernel.cu 0 additions, 132 deletionsoneflow/user/kernels/hardswish_kernel.cu
- oneflow/user/kernels/hardtanh_kernel.cpp 0 additions, 109 deletionsoneflow/user/kernels/hardtanh_kernel.cpp
- oneflow/user/kernels/hardtanh_kernel.cu 0 additions, 132 deletionsoneflow/user/kernels/hardtanh_kernel.cu
Please register or sign in to comment