845 lines
		
	
	
		
			22 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			845 lines
		
	
	
		
			22 KiB
		
	
	
	
		
			C++
		
	
	
	
| // Copyright 2019 Google LLC
 | |
| //
 | |
| // This source code is licensed under the BSD-style license found in the
 | |
| // LICENSE file in the root directory of this source tree.
 | |
| 
 | |
| #include <gtest/gtest.h>
 | |
| 
 | |
| #include <xnnpack/common.h>
 | |
| #include <xnnpack/isa-checks.h>
 | |
| 
 | |
| #include <xnnpack/pad.h>
 | |
| #include "pad-microkernel-tester.h"
 | |
| 
 | |
| 
 | |
| #if XNN_ARCH_ARM || XNN_ARCH_ARM64
 | |
|   TEST(XX_PAD__NEON, fulltile_copy_channels_eq_16) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(16)
 | |
|       .Test(xnn_xx_pad_ukernel__neon);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, fulltile_copy_channels_div_16) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     for (size_t channels = 32; channels <= 48; channels += 16) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(channels)
 | |
|         .Test(xnn_xx_pad_ukernel__neon);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, fulltile_copy_channels_lt_16) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     for (size_t channels = 1; channels < 16; channels++) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(channels)
 | |
|         .Test(xnn_xx_pad_ukernel__neon);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, fulltile_copy_channels_gt_16) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     for (size_t channels = 17; channels < 32; channels++) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(channels)
 | |
|         .Test(xnn_xx_pad_ukernel__neon);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, fulltile_pre_padding_eq_1) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .pre_padding(1)
 | |
|       .Test(xnn_xx_pad_ukernel__neon);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, fulltile_pre_padding_eq_2) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .pre_padding(2)
 | |
|       .Test(xnn_xx_pad_ukernel__neon);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, fulltile_pre_padding_eq_4) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .pre_padding(4)
 | |
|       .Test(xnn_xx_pad_ukernel__neon);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, fulltile_pre_padding_eq_16) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .pre_padding(16)
 | |
|       .Test(xnn_xx_pad_ukernel__neon);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, fulltile_pre_padding_div_16) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     for (size_t pre_padding = 32; pre_padding <= 48; pre_padding += 16) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(1)
 | |
|         .pre_padding(pre_padding)
 | |
|         .Test(xnn_xx_pad_ukernel__neon);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, fulltile_pre_padding_lt_16) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     for (size_t pre_padding = 1; pre_padding < 16; pre_padding++) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(1)
 | |
|         .pre_padding(pre_padding)
 | |
|         .Test(xnn_xx_pad_ukernel__neon);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, fulltile_pre_padding_gt_16) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     for (size_t pre_padding = 17; pre_padding < 32; pre_padding++) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(1)
 | |
|         .pre_padding(pre_padding)
 | |
|         .Test(xnn_xx_pad_ukernel__neon);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, fulltile_post_padding_eq_1) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .post_padding(1)
 | |
|       .Test(xnn_xx_pad_ukernel__neon);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, fulltile_post_padding_eq_2) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .post_padding(2)
 | |
|       .Test(xnn_xx_pad_ukernel__neon);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, fulltile_post_padding_eq_4) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .post_padding(4)
 | |
|       .Test(xnn_xx_pad_ukernel__neon);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, fulltile_post_padding_eq_16) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .post_padding(16)
 | |
|       .Test(xnn_xx_pad_ukernel__neon);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, fulltile_post_padding_div_16) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     for (size_t post_padding = 32; post_padding <= 48; post_padding += 16) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(1)
 | |
|         .post_padding(post_padding)
 | |
|         .Test(xnn_xx_pad_ukernel__neon);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, fulltile_post_padding_lt_16) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     for (size_t post_padding = 1; post_padding < 16; post_padding++) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(1)
 | |
|         .post_padding(post_padding)
 | |
|         .Test(xnn_xx_pad_ukernel__neon);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, fulltile_post_padding_gt_16) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     for (size_t post_padding = 17; post_padding < 32; post_padding++) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(1)
 | |
|         .pre_padding(post_padding)
 | |
|         .Test(xnn_xx_pad_ukernel__neon);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, multitile) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     for (size_t rows = 2; rows <= 5; rows++) {
 | |
|       for (size_t channels = 1; channels < 48; channels += 3) {
 | |
|         PadMicrokernelTester()
 | |
|           .rows(rows)
 | |
|           .input_channels(channels)
 | |
|           .pre_padding(channels)
 | |
|           .post_padding(channels)
 | |
|           .Test(xnn_xx_pad_ukernel__neon);
 | |
|       }
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, multitile_with_input_stride) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     for (size_t rows = 2; rows <= 5; rows++) {
 | |
|       for (size_t channels = 1; channels < 48; channels += 3) {
 | |
|         PadMicrokernelTester()
 | |
|           .rows(rows)
 | |
|           .input_channels(channels)
 | |
|           .pre_padding(channels)
 | |
|           .post_padding(channels)
 | |
|           .input_stride(51)
 | |
|           .Test(xnn_xx_pad_ukernel__neon);
 | |
|       }
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__NEON, multitile_with_output_stride) {
 | |
|     TEST_REQUIRES_ARM_NEON;
 | |
|     for (size_t rows = 2; rows <= 5; rows++) {
 | |
|       for (size_t channels = 1; channels < 48; channels += 3) {
 | |
|         PadMicrokernelTester()
 | |
|           .rows(rows)
 | |
|           .input_channels(2 * channels)
 | |
|           .pre_padding(channels)
 | |
|           .post_padding(channels)
 | |
|           .output_stride(193)
 | |
|           .Test(xnn_xx_pad_ukernel__neon);
 | |
|       }
 | |
|     }
 | |
|   }
 | |
| #endif  // XNN_ARCH_ARM || XNN_ARCH_ARM64
 | |
| 
 | |
| 
 | |
| #if XNN_ARCH_X86 || XNN_ARCH_X86_64
 | |
|   TEST(XX_PAD__SSE2, fulltile_copy_channels_eq_16) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(16)
 | |
|       .Test(xnn_xx_pad_ukernel__sse2);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, fulltile_copy_channels_div_16) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     for (size_t channels = 32; channels <= 48; channels += 16) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(channels)
 | |
|         .Test(xnn_xx_pad_ukernel__sse2);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, fulltile_copy_channels_lt_16) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     for (size_t channels = 1; channels < 16; channels++) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(channels)
 | |
|         .Test(xnn_xx_pad_ukernel__sse2);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, fulltile_copy_channels_gt_16) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     for (size_t channels = 17; channels < 32; channels++) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(channels)
 | |
|         .Test(xnn_xx_pad_ukernel__sse2);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, fulltile_pre_padding_eq_1) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .pre_padding(1)
 | |
|       .Test(xnn_xx_pad_ukernel__sse2);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, fulltile_pre_padding_eq_2) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .pre_padding(2)
 | |
|       .Test(xnn_xx_pad_ukernel__sse2);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, fulltile_pre_padding_eq_4) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .pre_padding(4)
 | |
|       .Test(xnn_xx_pad_ukernel__sse2);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, fulltile_pre_padding_eq_16) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .pre_padding(16)
 | |
|       .Test(xnn_xx_pad_ukernel__sse2);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, fulltile_pre_padding_div_16) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     for (size_t pre_padding = 32; pre_padding <= 48; pre_padding += 16) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(1)
 | |
|         .pre_padding(pre_padding)
 | |
|         .Test(xnn_xx_pad_ukernel__sse2);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, fulltile_pre_padding_lt_16) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     for (size_t pre_padding = 1; pre_padding < 16; pre_padding++) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(1)
 | |
|         .pre_padding(pre_padding)
 | |
|         .Test(xnn_xx_pad_ukernel__sse2);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, fulltile_pre_padding_gt_16) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     for (size_t pre_padding = 17; pre_padding < 32; pre_padding++) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(1)
 | |
|         .pre_padding(pre_padding)
 | |
|         .Test(xnn_xx_pad_ukernel__sse2);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, fulltile_post_padding_eq_1) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .post_padding(1)
 | |
|       .Test(xnn_xx_pad_ukernel__sse2);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, fulltile_post_padding_eq_2) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .post_padding(2)
 | |
|       .Test(xnn_xx_pad_ukernel__sse2);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, fulltile_post_padding_eq_4) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .post_padding(4)
 | |
|       .Test(xnn_xx_pad_ukernel__sse2);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, fulltile_post_padding_eq_16) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .post_padding(16)
 | |
|       .Test(xnn_xx_pad_ukernel__sse2);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, fulltile_post_padding_div_16) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     for (size_t post_padding = 32; post_padding <= 48; post_padding += 16) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(1)
 | |
|         .post_padding(post_padding)
 | |
|         .Test(xnn_xx_pad_ukernel__sse2);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, fulltile_post_padding_lt_16) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     for (size_t post_padding = 1; post_padding < 16; post_padding++) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(1)
 | |
|         .post_padding(post_padding)
 | |
|         .Test(xnn_xx_pad_ukernel__sse2);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, fulltile_post_padding_gt_16) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     for (size_t post_padding = 17; post_padding < 32; post_padding++) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(1)
 | |
|         .pre_padding(post_padding)
 | |
|         .Test(xnn_xx_pad_ukernel__sse2);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, multitile) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     for (size_t rows = 2; rows <= 5; rows++) {
 | |
|       for (size_t channels = 1; channels < 48; channels += 3) {
 | |
|         PadMicrokernelTester()
 | |
|           .rows(rows)
 | |
|           .input_channels(channels)
 | |
|           .pre_padding(channels)
 | |
|           .post_padding(channels)
 | |
|           .Test(xnn_xx_pad_ukernel__sse2);
 | |
|       }
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, multitile_with_input_stride) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     for (size_t rows = 2; rows <= 5; rows++) {
 | |
|       for (size_t channels = 1; channels < 48; channels += 3) {
 | |
|         PadMicrokernelTester()
 | |
|           .rows(rows)
 | |
|           .input_channels(channels)
 | |
|           .pre_padding(channels)
 | |
|           .post_padding(channels)
 | |
|           .input_stride(51)
 | |
|           .Test(xnn_xx_pad_ukernel__sse2);
 | |
|       }
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__SSE2, multitile_with_output_stride) {
 | |
|     TEST_REQUIRES_X86_SSE2;
 | |
|     for (size_t rows = 2; rows <= 5; rows++) {
 | |
|       for (size_t channels = 1; channels < 48; channels += 3) {
 | |
|         PadMicrokernelTester()
 | |
|           .rows(rows)
 | |
|           .input_channels(2 * channels)
 | |
|           .pre_padding(channels)
 | |
|           .post_padding(channels)
 | |
|           .output_stride(193)
 | |
|           .Test(xnn_xx_pad_ukernel__sse2);
 | |
|       }
 | |
|     }
 | |
|   }
 | |
| #endif  // XNN_ARCH_X86 || XNN_ARCH_X86_64
 | |
| 
 | |
| 
 | |
| #if XNN_ARCH_WASMSIMD
 | |
|   TEST(XX_PAD__WASMSIMD, fulltile_copy_channels_eq_16) {
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(16)
 | |
|       .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, fulltile_copy_channels_div_16) {
 | |
|     for (size_t channels = 32; channels <= 48; channels += 16) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(channels)
 | |
|         .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, fulltile_copy_channels_lt_16) {
 | |
|     for (size_t channels = 1; channels < 16; channels++) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(channels)
 | |
|         .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, fulltile_copy_channels_gt_16) {
 | |
|     for (size_t channels = 17; channels < 32; channels++) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(channels)
 | |
|         .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, fulltile_pre_padding_eq_1) {
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .pre_padding(1)
 | |
|       .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, fulltile_pre_padding_eq_2) {
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .pre_padding(2)
 | |
|       .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, fulltile_pre_padding_eq_4) {
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .pre_padding(4)
 | |
|       .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, fulltile_pre_padding_eq_16) {
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .pre_padding(16)
 | |
|       .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, fulltile_pre_padding_div_16) {
 | |
|     for (size_t pre_padding = 32; pre_padding <= 48; pre_padding += 16) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(1)
 | |
|         .pre_padding(pre_padding)
 | |
|         .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, fulltile_pre_padding_lt_16) {
 | |
|     for (size_t pre_padding = 1; pre_padding < 16; pre_padding++) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(1)
 | |
|         .pre_padding(pre_padding)
 | |
|         .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, fulltile_pre_padding_gt_16) {
 | |
|     for (size_t pre_padding = 17; pre_padding < 32; pre_padding++) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(1)
 | |
|         .pre_padding(pre_padding)
 | |
|         .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, fulltile_post_padding_eq_1) {
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .post_padding(1)
 | |
|       .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, fulltile_post_padding_eq_2) {
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .post_padding(2)
 | |
|       .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, fulltile_post_padding_eq_4) {
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .post_padding(4)
 | |
|       .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, fulltile_post_padding_eq_16) {
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .post_padding(16)
 | |
|       .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, fulltile_post_padding_div_16) {
 | |
|     for (size_t post_padding = 32; post_padding <= 48; post_padding += 16) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(1)
 | |
|         .post_padding(post_padding)
 | |
|         .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, fulltile_post_padding_lt_16) {
 | |
|     for (size_t post_padding = 1; post_padding < 16; post_padding++) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(1)
 | |
|         .post_padding(post_padding)
 | |
|         .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, fulltile_post_padding_gt_16) {
 | |
|     for (size_t post_padding = 17; post_padding < 32; post_padding++) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(1)
 | |
|         .input_channels(1)
 | |
|         .pre_padding(post_padding)
 | |
|         .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, multitile) {
 | |
|     for (size_t rows = 2; rows <= 5; rows++) {
 | |
|       for (size_t channels = 1; channels < 48; channels += 3) {
 | |
|         PadMicrokernelTester()
 | |
|           .rows(rows)
 | |
|           .input_channels(channels)
 | |
|           .pre_padding(channels)
 | |
|           .post_padding(channels)
 | |
|           .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|       }
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, multitile_with_input_stride) {
 | |
|     for (size_t rows = 2; rows <= 5; rows++) {
 | |
|       for (size_t channels = 1; channels < 48; channels += 3) {
 | |
|         PadMicrokernelTester()
 | |
|           .rows(rows)
 | |
|           .input_channels(channels)
 | |
|           .pre_padding(channels)
 | |
|           .post_padding(channels)
 | |
|           .input_stride(51)
 | |
|           .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|       }
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   TEST(XX_PAD__WASMSIMD, multitile_with_output_stride) {
 | |
|     for (size_t rows = 2; rows <= 5; rows++) {
 | |
|       for (size_t channels = 1; channels < 48; channels += 3) {
 | |
|         PadMicrokernelTester()
 | |
|           .rows(rows)
 | |
|           .input_channels(2 * channels)
 | |
|           .pre_padding(channels)
 | |
|           .post_padding(channels)
 | |
|           .output_stride(193)
 | |
|           .Test(xnn_xx_pad_ukernel__wasmsimd);
 | |
|       }
 | |
|     }
 | |
|   }
 | |
| #endif  // XNN_ARCH_WASMSIMD
 | |
| 
 | |
| 
 | |
| TEST(XX_PAD__SCALAR, fulltile_copy_channels_eq_16) {
 | |
|   PadMicrokernelTester()
 | |
|     .rows(1)
 | |
|     .input_channels(16)
 | |
|     .Test(xnn_xx_pad_ukernel__scalar);
 | |
| }
 | |
| 
 | |
| TEST(XX_PAD__SCALAR, fulltile_copy_channels_div_16) {
 | |
|   for (size_t channels = 32; channels <= 48; channels += 16) {
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(channels)
 | |
|       .Test(xnn_xx_pad_ukernel__scalar);
 | |
|   }
 | |
| }
 | |
| 
 | |
| TEST(XX_PAD__SCALAR, fulltile_copy_channels_lt_16) {
 | |
|   for (size_t channels = 1; channels < 16; channels++) {
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(channels)
 | |
|       .Test(xnn_xx_pad_ukernel__scalar);
 | |
|   }
 | |
| }
 | |
| 
 | |
| TEST(XX_PAD__SCALAR, fulltile_copy_channels_gt_16) {
 | |
|   for (size_t channels = 17; channels < 32; channels++) {
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(channels)
 | |
|       .Test(xnn_xx_pad_ukernel__scalar);
 | |
|   }
 | |
| }
 | |
| 
 | |
| TEST(XX_PAD__SCALAR, fulltile_pre_padding_eq_1) {
 | |
|   PadMicrokernelTester()
 | |
|     .rows(1)
 | |
|     .input_channels(1)
 | |
|     .pre_padding(1)
 | |
|     .Test(xnn_xx_pad_ukernel__scalar);
 | |
| }
 | |
| 
 | |
| TEST(XX_PAD__SCALAR, fulltile_pre_padding_eq_2) {
 | |
|   PadMicrokernelTester()
 | |
|     .rows(1)
 | |
|     .input_channels(1)
 | |
|     .pre_padding(2)
 | |
|     .Test(xnn_xx_pad_ukernel__scalar);
 | |
| }
 | |
| 
 | |
| TEST(XX_PAD__SCALAR, fulltile_pre_padding_eq_4) {
 | |
|   PadMicrokernelTester()
 | |
|     .rows(1)
 | |
|     .input_channels(1)
 | |
|     .pre_padding(4)
 | |
|     .Test(xnn_xx_pad_ukernel__scalar);
 | |
| }
 | |
| 
 | |
| TEST(XX_PAD__SCALAR, fulltile_pre_padding_div_4) {
 | |
|   for (size_t pre_padding = 8; pre_padding <= 12; pre_padding += 4) {
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .pre_padding(pre_padding)
 | |
|       .Test(xnn_xx_pad_ukernel__scalar);
 | |
|   }
 | |
| }
 | |
| 
 | |
| TEST(XX_PAD__SCALAR, fulltile_pre_padding_lt_4) {
 | |
|   for (size_t pre_padding = 1; pre_padding < 4; pre_padding++) {
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .pre_padding(pre_padding)
 | |
|       .Test(xnn_xx_pad_ukernel__scalar);
 | |
|   }
 | |
| }
 | |
| 
 | |
| TEST(XX_PAD__SCALAR, fulltile_pre_padding_gt_4) {
 | |
|   for (size_t pre_padding = 5; pre_padding < 8; pre_padding++) {
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .pre_padding(pre_padding)
 | |
|       .Test(xnn_xx_pad_ukernel__scalar);
 | |
|   }
 | |
| }
 | |
| 
 | |
| TEST(XX_PAD__SCALAR, fulltile_post_padding_eq_1) {
 | |
|   PadMicrokernelTester()
 | |
|     .rows(1)
 | |
|     .input_channels(1)
 | |
|     .post_padding(1)
 | |
|     .Test(xnn_xx_pad_ukernel__scalar);
 | |
| }
 | |
| 
 | |
| TEST(XX_PAD__SCALAR, fulltile_post_padding_eq_2) {
 | |
|   PadMicrokernelTester()
 | |
|     .rows(1)
 | |
|     .input_channels(1)
 | |
|     .post_padding(2)
 | |
|     .Test(xnn_xx_pad_ukernel__scalar);
 | |
| }
 | |
| 
 | |
| TEST(XX_PAD__SCALAR, fulltile_post_padding_eq_4) {
 | |
|   PadMicrokernelTester()
 | |
|     .rows(1)
 | |
|     .input_channels(1)
 | |
|     .post_padding(4)
 | |
|     .Test(xnn_xx_pad_ukernel__scalar);
 | |
| }
 | |
| 
 | |
| TEST(XX_PAD__SCALAR, fulltile_post_padding_div_4) {
 | |
|   for (size_t post_padding = 8; post_padding <= 12; post_padding += 4) {
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .post_padding(post_padding)
 | |
|       .Test(xnn_xx_pad_ukernel__scalar);
 | |
|   }
 | |
| }
 | |
| 
 | |
| TEST(XX_PAD__SCALAR, fulltile_post_padding_lt_4) {
 | |
|   for (size_t post_padding = 1; post_padding < 4; post_padding++) {
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .post_padding(post_padding)
 | |
|       .Test(xnn_xx_pad_ukernel__scalar);
 | |
|   }
 | |
| }
 | |
| 
 | |
| TEST(XX_PAD__SCALAR, fulltile_post_padding_gt_4) {
 | |
|   for (size_t post_padding = 5; post_padding < 8; post_padding++) {
 | |
|     PadMicrokernelTester()
 | |
|       .rows(1)
 | |
|       .input_channels(1)
 | |
|       .pre_padding(post_padding)
 | |
|       .Test(xnn_xx_pad_ukernel__scalar);
 | |
|   }
 | |
| }
 | |
| 
 | |
| TEST(XX_PAD__SCALAR, multitile) {
 | |
|   for (size_t rows = 2; rows <= 5; rows++) {
 | |
|     for (size_t channels = 1; channels < 12; channels += 3) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(rows)
 | |
|         .input_channels(channels)
 | |
|         .pre_padding(channels)
 | |
|         .post_padding(channels)
 | |
|         .Test(xnn_xx_pad_ukernel__scalar);
 | |
|     }
 | |
|   }
 | |
| }
 | |
| 
 | |
| TEST(XX_PAD__SCALAR, multitile_with_input_stride) {
 | |
|   for (size_t rows = 2; rows <= 5; rows++) {
 | |
|     for (size_t channels = 1; channels < 12; channels += 3) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(rows)
 | |
|         .input_channels(channels)
 | |
|         .pre_padding(channels)
 | |
|         .post_padding(channels)
 | |
|         .input_stride(51)
 | |
|         .Test(xnn_xx_pad_ukernel__scalar);
 | |
|     }
 | |
|   }
 | |
| }
 | |
| 
 | |
| TEST(XX_PAD__SCALAR, multitile_with_output_stride) {
 | |
|   for (size_t rows = 2; rows <= 5; rows++) {
 | |
|     for (size_t channels = 1; channels < 12; channels += 3) {
 | |
|       PadMicrokernelTester()
 | |
|         .rows(rows)
 | |
|         .input_channels(2 * channels)
 | |
|         .pre_padding(channels)
 | |
|         .post_padding(channels)
 | |
|         .output_stride(193)
 | |
|         .Test(xnn_xx_pad_ukernel__scalar);
 | |
|     }
 | |
|   }
 | |
| }
 |