| 7 | using namespace megcc::KernelGen; |
| 8 | |
| 9 | TEST(AARCH64, Conv1x1NCHW44) { |
| 10 | Checker<ConvolutionForward> checker(Arch::ARM64); |
| 11 | checker.set_kernel_symbol("Arm64_kernel_conv2d_conv1x1_NCHW44.+"); |
| 12 | ConvolutionForward::Param param; |
| 13 | param.pad_h = 0; |
| 14 | param.pad_w = 0; |
| 15 | param.stride_h = 1; |
| 16 | param.stride_w = 1; |
| 17 | param.compute_mode = ConvolutionForward::Param::ComputeMode::DEFAULT; |
| 18 | param.format = ConvolutionForward::Param::Format::NCHW44; |
| 19 | checker.set_param(param); |
| 20 | checker.execs({{2, 3, 5, 11, 4}, {3, 3, 1, 1, 4, 4}, {}}); |
| 21 | |
| 22 | param.sparse = ConvolutionForward::Param::Sparse::GROUP; |
| 23 | checker.set_param(param); |
| 24 | checker.execs({{2, 6, 17, 19, 4}, {2, 4, 3, 1, 1, 4, 4}, {}}); |
| 25 | } |
| 26 | #if !MEGCC_WITHOUT_DOT |
| 27 | TEST(AARCH64, ConvBias1x1NCHW44Dot) { |
| 28 | Checker<ConvBiasForward> checker(Arch::ARM64); |