Blame view

3rdparty/opencv-4.5.4/cmake/checks/runtime/cpu_vsx_aligned.cpp 1.2 KB
f4334277   Hu Chunming   提交3rdparty
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
  // check sanity of vsx aligned ld/st
  // https://github.com/opencv/opencv/issues/13211
  
  #include <altivec.h>
  #undef bool
  
  #define vsx_ld vec_vsx_ld
  #define vsx_st vec_vsx_st
  
  template<typename T>
  static void fill(T& d, int from = 0, int to = 16)
  {
     for (int i = from; i < to; i++)
          d[i] = i;
  }
  
  template<typename T, typename Tvec>
  static bool check_data(T& d, Tvec& v, int from = 0, int to = 16)
  {
      for (int i = from; i < to; i++)
      {
          if (d[i] != vec_extract(v, i))
              return false;
      }
      return true;
  }
  
  int main()
  {
      unsigned char __attribute__ ((aligned (16))) rbuf[16];
      unsigned char __attribute__ ((aligned (16))) wbuf[16];
      __vector unsigned char a;
  
      // 1- check aligned load and store
      fill(rbuf);
      a = vec_ld(0, rbuf);
      if (!check_data(rbuf, a))
          return 1;
      vec_st(a, 0, wbuf);
      if (!check_data(wbuf, a))
          return 11;
  
      // 2- check mixing aligned load and unaligned store
      a = vec_ld(0, rbuf);
      vsx_st(a, 0, wbuf);
      if (!check_data(wbuf, a))
          return 2;
  
      // 3- check mixing unaligned load and aligned store
      a = vsx_ld(0, rbuf);
      vec_st(a, 0, wbuf);
      if (!check_data(wbuf, a))
          return 3;
  
      return 0;
  }