| 5 | using namespace megcc::KernelGen; |
| 6 | |
| 7 | TEST(AARCH64, CVresize) { |
| 8 | Checker<megdnn::CVResize> checker(Arch::ARM64); |
| 9 | megdnn::CVResize::Param param; |
| 10 | param.format = megdnn::CVResize::Param::Format::NHWC; |
| 11 | param.imode = megdnn::CVResize::Param::InterpolationMode::LINEAR; |
| 12 | UniformIntRNG rng(0, 254); |
| 13 | checker.set_rng(0, &rng); |
| 14 | checker.set_param(param); |
| 15 | checker.set_dtype(0, dtype::Uint8()); |
| 16 | checker.set_dtype(1, dtype::Uint8()); |
| 17 | checker.set_epsilon(1 + 1e-4); |
| 18 | for (size_t h : {2, 3, 5, 6, 7, 9, 10}) |
| 19 | for (size_t w : {2, 3, 5, 7, 8, 9, 10, 11}) { |
| 20 | checker.exec({{1, h, w, 3}, {1, w, h, 3}}); |
| 21 | checker.exec({{1, h, w, 1}, {1, w, h, 1}}); |
| 22 | checker.exec({{1, h, w, 2}, {1, w, h, 2}}); |
| 23 | checker.exec({{1, h, w, 5}, {1, w, h, 5}}); |
| 24 | } |
| 25 | checker.exec({{1, 3, 5, 3}, {1, 6, 8, 3}}); |
| 26 | checker.exec({{1, 13, 11, 1}, {1, 3, 8, 1}}); |
| 27 | } |
| 28 | |
| 29 | TEST(AARCH64, CVcvtcolor) { |
| 30 | Checker<megdnn::CVCvtColor> checker(Arch::ARM64); |