|
|
@@ -25,12 +25,28 @@ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(ERFCINV, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(H_SWISH, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(SILU, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(GELU, cb) |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(GELU, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(SINH, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(COSH, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(ASINH, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(ACOSH, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(ATANH, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(TAN, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(SOFTPLUS, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(RELU6, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(HSIGMOID, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(LOGSIGMOID, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(SQRT, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(SQUARE, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(SIGN, cb) |
|
|
|
|
|
|
|
#define MEGDNN_FOREACH_ELEMWISE_MODE_UNARY_INT(cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(RELU, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(ABS, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(NEGATE, cb) |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(NEGATE, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(RELU6, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(SQUARE, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(SIGN, cb) |
|
|
|
|
|
|
|
#define MEGDNN_FOREACH_ELEMWISE_MODE_BINARY_BOOL(cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(AND, cb) \ |
|
|
@@ -66,7 +82,14 @@ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(H_SWISH_GRAD, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(FUSE_ADD_H_SWISH, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(SILU_GRAD, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(GELU_GRAD, cb) |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(GELU_GRAD, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(PRELU, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(ASINH_GRAD, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(ACOSH_GRAD, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(ATANH_GRAD, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(SOFTPLUS_GRAD, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(RELU6_GRAD, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(HSIGMOID_GRAD, cb) |
|
|
|
|
|
|
|
#define MEGDNN_FOREACH_ELEMWISE_MODE_BINARY_INT(cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(ABS_GRAD, cb) \ |
|
|
@@ -86,15 +109,19 @@ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(FUSE_ADD_RELU, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(SHL, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(SHR, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(RMULH, cb) |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(RMULH, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(PRELU, cb) |
|
|
|
|
|
|
|
#define MEGDNN_FOREACH_ELEMWISE_MODE_TERNARY_BOOL(cb) |
|
|
|
|
|
|
|
#define MEGDNN_FOREACH_ELEMWISE_MODE_TERNARY_FLOAT(cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(COND_LEQ_MOV, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(COND_LT_MOV, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(FUSE_MUL_ADD3, cb) |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(FUSE_MUL_ADD3, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(CLIP, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(PRELU_GRAD, cb) |
|
|
|
|
|
|
|
#define MEGDNN_FOREACH_ELEMWISE_MODE_TERNARY_INT(cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(COND_LEQ_MOV, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(COND_LT_MOV, cb) |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(COND_LT_MOV, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(CLIP, cb) |