[4/5] add utest compiler_skip_check for extension cl_intel_device_side_avc_motion_estimation.

Submitted by Luo, Xionghu on June 13, 2017, 4:54 p.m.

Details

Message ID 1497372856-831-4-git-send-email-xionghu.luo@intel.com
State New
Headers show
Series "Series without cover letter" ( rev: 1 ) in Beignet

Browsing this patch as part of:
"Series without cover letter" rev 1 in Beignet
<< prev patch [4/5] next patch >>

Commit Message

Luo, Xionghu June 13, 2017, 4:54 p.m.
From: Luo Xionghu <xionghu.luo@intel.com>

fix build warnings.
Signed-off-by: Chuanbo Weng <chuanbo.weng@intel.com>
Signed-off-by: Xionghu Luo <xionghu.luo@intel.com>
---
 kernels/compiler_skip_check.cl |  53 ++++++++++++
 utests/CMakeLists.txt          |   1 +
 utests/compiler_skip_check.cpp | 190 +++++++++++++++++++++++++++++++++++++++++
 3 files changed, 244 insertions(+)
 create mode 100644 kernels/compiler_skip_check.cl
 create mode 100644 utests/compiler_skip_check.cpp

Patch hide | download patch | download mbox

diff --git a/kernels/compiler_skip_check.cl b/kernels/compiler_skip_check.cl
new file mode 100644
index 0000000..68369dd
--- /dev/null
+++ b/kernels/compiler_skip_check.cl
@@ -0,0 +1,53 @@ 
+__kernel __attribute__((intel_reqd_sub_group_size(16)))
+void compiler_skip_check(__read_only image2d_t   src_img,
+                         __read_only image2d_t   ref_img,
+                         __global short2  *motion_vector_buffer,
+                         __global ushort  *residual_buffer,
+                         __global uint*          dwo_buffer,
+                         __global uint*          pld_buffer){
+  sampler_t vs = 0;
+
+  int gr_id0 = get_group_id(0);
+  int gr_id1 = get_group_id(1);
+
+  ushort2 src_coord;
+
+  src_coord.x = gr_id0 * 16;
+  src_coord.y = gr_id1 * 16;
+
+  intel_sub_group_avc_sic_payload_t payload =
+      intel_sub_group_avc_sic_initialize(src_coord);
+
+  //Ignore in unidirectional, so just set to 0
+  uchar bidir_weight = 0;
+
+  uint skip_block_partition_type = CLK_AVC_ME_SKIP_BLOCK_PARTITION_16x16_INTEL;
+  uint skip_motion_vector_mask = CLK_AVC_ME_SKIP_BLOCK_16x16_FORWARD_ENABLE_INTEL;
+  uchar skip_sad_adjustment = CLK_AVC_ME_SAD_ADJUST_MODE_NONE_INTEL;
+
+  uint2 bi_mv;
+  int mb_idx =  gr_id0  +  gr_id1 * get_num_groups(0);
+  short2 input_mv = motion_vector_buffer[mb_idx];
+  bi_mv.s0 = as_uint(input_mv);
+
+  ulong mv = as_ulong(bi_mv);
+
+  payload = intel_sub_group_avc_sic_configure_skc(
+      skip_block_partition_type, skip_motion_vector_mask, mv, bidir_weight,
+      skip_sad_adjustment, payload);
+
+  intel_sub_group_avc_sic_result_t result =
+      intel_sub_group_avc_sic_evaluate_with_single_reference(
+          src_img, ref_img, vs, payload);
+
+  ushort distortion = intel_sub_group_avc_sic_get_inter_distortions(result);
+
+  int lid_x = get_local_id(0);
+  if(lid_x == 0)
+    residual_buffer[mb_idx] = distortion;
+  dwo_buffer[mb_idx*16*4 + lid_x + 16*0] = result.s0;
+  dwo_buffer[mb_idx*16*4 + lid_x + 16*1] = result.s1;
+  dwo_buffer[mb_idx*16*4 + lid_x + 16*2] = result.s2;
+  dwo_buffer[mb_idx*16*4 + lid_x + 16*3] = result.s3;
+
+}
diff --git a/utests/CMakeLists.txt b/utests/CMakeLists.txt
index 2bdae84..61d1326 100644
--- a/utests/CMakeLists.txt
+++ b/utests/CMakeLists.txt
@@ -306,6 +306,7 @@  set (utests_sources
   compiler_device_enqueue.cpp
   compiler_sqrt_div.cpp
   compiler_remove_negative_add.cpp
+  compiler_skip_check.cpp
   compiler_intra_prediction.cpp)
 
 if (LLVM_VERSION_NODOT VERSION_GREATER 34)
diff --git a/utests/compiler_skip_check.cpp b/utests/compiler_skip_check.cpp
new file mode 100644
index 0000000..cb4b290
--- /dev/null
+++ b/utests/compiler_skip_check.cpp
@@ -0,0 +1,190 @@ 
+#include "utest_helper.hpp"
+#include <string.h>
+
+const size_t w = 80;
+const size_t h = 48;
+const size_t mv_w = (w + 15) / 16;
+const size_t mv_h = (h + 15) / 16;
+
+void cpu_result(uint8_t* srcImg, uint8_t* refImg, int16_t* mv, uint16_t* residual){
+  for (uint32_t j = 0; j <= mv_h - 1; ++j) {
+    for (uint32_t i = 0; i <= mv_w - 1; ++i) {
+      uint32_t mv_num = j * mv_w + i;
+      int16_t mv_x = mv[mv_num*2] >> 2;
+      int16_t mv_y = mv[mv_num*2 + 1] >> 2;
+      int16_t src_mb_x = i * 16;
+      int16_t src_mb_y = j * 16;
+      int16_t ref_mb_x = src_mb_x + mv_x;
+      int16_t ref_mb_y = src_mb_y + mv_y;
+
+      uint16_t res = 0;
+      int16_t sy = src_mb_y, ry = ref_mb_y;
+      for(uint32_t a = 0; a < 16; a++){
+        int16_t sx = src_mb_x;
+        int16_t rx = ref_mb_x;
+        for(uint32_t b = 0; b < 16; b++){
+          uint8_t src_pixel = srcImg[sy*w + sx];
+          uint8_t ref_pixel = refImg[ry*w + rx];
+          res += abs(src_pixel - ref_pixel);
+          sx++;
+          rx++;
+        }
+        sy++;
+        ry++;
+      }
+      residual[mv_num] = res;
+    }
+  }
+}
+
+void compiler_skip_check(void)
+{
+  if (!cl_check_device_side_avc_motion_estimation()) {
+    return;
+  }
+
+  OCL_CREATE_KERNEL("compiler_skip_check");
+
+  cl_image_format format;
+  cl_image_desc desc;
+
+  memset(&desc, 0x0, sizeof(cl_image_desc));
+  memset(&format, 0x0, sizeof(cl_image_format));
+
+  uint8_t* image_data1 = (uint8_t *)malloc(w * h);    //src
+  uint8_t* image_data2 = (uint8_t *)malloc(w * h);    //ref
+  for (size_t j = 0; j < h; j++) {
+    for (size_t i = 0; i < w; i++) {
+      if (i >= 32 && i <= 47 && j >= 16 && j <= 31)
+        image_data1[w * j + i] = 100;
+      else
+        image_data1[w * j + i] = j + i;
+      if (i >= 33 && i <= 48 && j >= 14 && j <= 29)
+        image_data2[w * j + i] = 99;
+      else
+        image_data2[w * j + i] = (h-1) + (w-1) - (j+i);
+    }
+  }
+
+
+  format.image_channel_order = CL_R;
+  format.image_channel_data_type = CL_UNORM_INT8;
+  desc.image_type = CL_MEM_OBJECT_IMAGE2D;
+  desc.image_width = w;
+  desc.image_height = h;
+  desc.image_row_pitch = 0;
+  OCL_CREATE_IMAGE(buf[0], CL_MEM_COPY_HOST_PTR, &format, &desc, image_data1);        //src
+  OCL_CREATE_IMAGE(buf[1], CL_MEM_COPY_HOST_PTR, &format, &desc, image_data2);        //ref
+
+  int16_t* input_mv = (int16_t*) malloc(mv_w * mv_h * sizeof(int16_t) * 2);
+  //Generate input mv data
+  for (uint32_t j = 0; j <= mv_h - 1; ++j) {
+    for (uint32_t i = 0; i <= mv_w - 1; ++i) {
+      uint32_t mv_num = j * mv_w + i;
+      if(i == 32/16 && j == 16/16){
+        input_mv[mv_num*2 ] = 1;
+        input_mv[mv_num*2 + 1 ] = -2;
+      }
+      else{
+        input_mv[mv_num*2 ] = (mv_num) % 2;
+        input_mv[mv_num*2 + 1 ] = (mv_num) % 3;
+        if(i == mv_w - 1)
+          input_mv[mv_num*2 ] *= -1;
+        if(j == mv_h - 1)
+          input_mv[mv_num*2 + 1 ] *= -1;
+      }
+      input_mv[mv_num*2 ] <<= 2;
+      input_mv[mv_num*2 + 1 ] <<= 2;
+    }
+  }
+
+  uint16_t* cpu_resi = (uint16_t*) malloc(mv_w * mv_h * sizeof(uint16_t));
+  cpu_result(image_data1, image_data2, input_mv, cpu_resi);
+
+  OCL_CREATE_BUFFER(buf[2], CL_MEM_COPY_HOST_PTR, mv_w * mv_h * sizeof(int16_t) * 2, input_mv);
+  OCL_CREATE_BUFFER(buf[3], 0, mv_w * mv_h * sizeof(uint16_t), NULL);
+  OCL_CREATE_BUFFER(buf[4], 0, mv_w * mv_h * sizeof(uint32_t) * 16 * 8, NULL);
+  OCL_CREATE_BUFFER(buf[5], 0, mv_w * mv_h * sizeof(uint32_t) * 8 * 8, NULL);
+
+  OCL_SET_ARG(0, sizeof(cl_mem), &buf[0]);
+  OCL_SET_ARG(1, sizeof(cl_mem), &buf[1]);
+  OCL_SET_ARG(2, sizeof(cl_mem), &buf[2]);
+  OCL_SET_ARG(3, sizeof(cl_mem), &buf[3]);
+  OCL_SET_ARG(4, sizeof(cl_mem), &buf[4]);
+  OCL_SET_ARG(5, sizeof(cl_mem), &buf[5]);
+
+  globals[0] = w;
+  globals[1] = h / 16;
+  locals[0] = 16;
+  locals[1] = 1;
+  OCL_NDRANGE(2);
+
+  OCL_MAP_BUFFER(3);
+  OCL_MAP_BUFFER(4);
+  OCL_MAP_BUFFER(5);
+  uint16_t* residual = (uint16_t*)buf_data[3];
+#define VME_DEBUG 0
+#if VME_DEBUG
+  uint32_t* dwo = (uint32_t*)buf_data[4];
+  uint32_t* pld = (uint32_t*)buf_data[5];
+  std::cout<<std::endl;
+  for (uint32_t j = 0; j <= mv_h - 1; ++j) {
+    for (uint32_t i = 0; i <= mv_w - 1; ++i) {
+        uint32_t mv_num = j * mv_w + i;
+        std::cout<<"******* mv num = "<<mv_num<<": "<<std::endl;
+        std::cout<<"payload register result: "<<std::endl;
+        for(uint32_t row_num = 0; row_num < 8; row_num++){
+          for(int32_t idx = 7; idx >= 0; idx--)
+            printf("%.8x ", pld[mv_num*64 + row_num*8 + idx]);
+          printf("\n");
+        }
+        std::cout<<std::endl;
+        std::cout<<"writeback register result: "<<std::endl;
+        for(uint32_t row_num = 0; row_num < 4; row_num++){
+          for(int32_t wi = 7; wi >= 0; wi--)
+            printf("%.8x ", dwo[mv_num*16*4 + row_num*16 + wi]);
+          printf("\n");
+          for(int32_t wi = 15; wi >= 8; wi--)
+            printf("%.8x ", dwo[mv_num*16*4 + row_num*16 + wi]);
+          printf("\n");
+        }
+        std::cout<<std::endl;
+        std::cout<<"residual: ";
+        std::cout<<residual[mv_num]<<std::endl;
+    }
+  }
+  std::cout<<"cpu residual: "<<std::endl;
+  for (uint32_t j = 0; j <= mv_h - 1; ++j) {
+    for (uint32_t i = 0; i <= mv_w - 1; ++i) {
+        uint32_t mv_num = j * mv_w + i;
+        std::cout<<cpu_resi[mv_num]<<" ";
+    }
+  }
+  std::cout<<std::endl;
+  std::cout<<"gpu residual: "<<std::endl;
+  for (uint32_t j = 0; j <= mv_h - 1; ++j) {
+    for (uint32_t i = 0; i <= mv_w - 1; ++i) {
+        uint32_t mv_num = j * mv_w + i;
+        std::cout<<residual[mv_num]<<" ";
+    }
+  }
+#endif
+  std::cout<<std::endl;
+  for (uint32_t j = 0; j <= mv_h - 1; ++j) {
+    for (uint32_t i = 0; i <= mv_w - 1; ++i) {
+        uint32_t mv_num = j * mv_w + i;
+        OCL_ASSERT(cpu_resi[mv_num] == residual[mv_num]);
+    }
+  }
+
+  OCL_UNMAP_BUFFER(3);
+  OCL_UNMAP_BUFFER(4);
+  OCL_UNMAP_BUFFER(5);
+
+  free(image_data1);
+  free(image_data2);
+  free(input_mv);
+  free(cpu_resi);
+}
+
+MAKE_UTEST_FROM_FUNCTION(compiler_skip_check);