| 7 | using MODE = ElemwiseForward::Param::Mode; |
| 8 | |
| 9 | TEST(AARCH64, ElementwiseUnique) { |
| 10 | Checker<ElemwiseForward> checker(Arch::ARM64); |
| 11 | checker.set_kernel_symbol("ArmCommon_kernel_elementwise.+"); |
| 12 | ElemwiseForward::Param param; |
| 13 | for (auto mode : {MODE::RELU, MODE::EXP, MODE::SIGMOID, MODE::H_SWISH}) { |
| 14 | param.mode = mode; |
| 15 | checker.set_param(param); |
| 16 | checker.execs({{1, 10}, {}}); |
| 17 | checker.execs({{1, 10, 12, 13}, {}}); |
| 18 | checker.execs({{10, 8, 2, 1}, {}}); |
| 19 | } |
| 20 | } |
| 21 | |
| 22 | TEST(AARCH64, ElementwiseUnique_asm) { |
| 23 | Checker<ElemwiseForward> checker(Arch::ARM64); |
nothing calls this directly
no test coverage detected