4

我正在尝试优化此功能:

bool interpolate(const Mat &im, float ofsx, float ofsy, float a11, float a12, float a21, float a22, Mat &res)
{         
   bool ret = false;
   // input size (-1 for the safe bilinear interpolation)
   const int width = im.cols-1;
   const int height = im.rows-1;
   // output size
   const int halfWidth  = res.cols >> 1;
   const int halfHeight = res.rows >> 1;
   float *out = res.ptr<float>(0);
   const float *imptr  = im.ptr<float>(0);
   for (int j=-halfHeight; j<=halfHeight; ++j)
   {
      const float rx = ofsx + j * a12;
      const float ry = ofsy + j * a22;
      #pragma omp simd
      for(int i=-halfWidth; i<=halfWidth; ++i, out++)
      {
         float wx = rx + i * a11;
         float wy = ry + i * a21;
         const int x = (int) floor(wx);
         const int y = (int) floor(wy);
         if (x >= 0 && y >= 0 && x < width && y < height)
         {
            // compute weights
            wx -= x; wy -= y;
            int rowOffset = y*im.cols;
            int rowOffset1 = (y+1)*im.cols;
            // bilinear interpolation
            *out =
                (1.0f - wy) *
                ((1.0f - wx) * 
                imptr[rowOffset+x] +
                wx * 
                imptr[rowOffset+x+1]) +
                (       wy) *
                ((1.0f - wx) * 
                imptr[rowOffset1+x] + 
                wx *
                imptr[rowOffset1+x+1]);
         } else {
            *out = 0;
            ret =  true; // touching boundary of the input            
         }
      }
   }
   return ret;
}

我正在使用 Intel Advisor 对其进行优化,即使内部for已被矢量化,Intel Advisor 仍检测到低效的内存访问模式:

  • 60% 的单位/零步幅访问
  • 40% 的不规则/随机步幅访问

特别是在以下三个指令中有 4 个收集(不规则)访问:

在此处输入图像描述

根据我的理解,当访问的元素是 type 时,会发生收集访问的问题a[b],其中b是不可预测的。这似乎是 的情况imptr[rowOffset+x],其中rowOffsetx都是不可预测的。

同时,我看到Vertical Invariant当以恒定偏移量访问元素时,这应该发生(再次,根据我的理解)。但实际上我看不到这个常量偏移量在哪里

所以我有3个问题:

  1. 我是否正确理解了收集访问的问题?
  2. 垂直不变访问呢?我不太确定这一点。
  3. 最后,我怎样才能改善/解决这里的内存访问?

与 2017 更新 3 一起编译,icpc带有以下标志:

INTEL_OPT=-O3 -ipo -simd -xCORE-AVX2 -parallel -qopenmp -fargument-noalias -ansi-alias -no-prec-div -fp-model fast=2 -fma -align -finline-functions
INTEL_PROFILE=-g -qopt-report=5 -Bdynamic -shared-intel -debug inline-debug-info -qopenmp-link dynamic -parallel-source-info=2 -ldl
4

1 回答 1

1

向量化(SIMD 化)您的代码不会自动使您的访问模式更好(或更差)。为了最大限度地提高向量化代码的性能,您必须尝试在代码中使用单位步幅(也称为连续、线性、步幅-1)内存访问模式。或者至少是“可预测的”常规步幅-N,其中 N 在理想情况下应该是适中的低值。

在不引入这种规律性的情况下 - 您可以在指令级别保持内存 LOAD/STORE 操作部分顺序(非并行)。所以每次你想做“并行”加法/乘法等,你必须做“非并行”原始数据元素“收集”。

在您的情况下,似乎有常规的 stride-N(逻辑上) - 这可以从代码片段和 Advisor MAP 输出(在右侧面板上)中看到。 垂直不变- 意味着您有时会在迭代之间访问相同的内存位置。单位跨度意味着在其他情况下您具有逻辑上连续的内存访问权限。

但是,代码结构很复杂:循环体中有 if 语句,有复杂的条件和浮点 --> 整数(简单但仍然)转换。

因此,编译器必须使用最通用和最低效的方法(收集)“以防万一”,因此您的物理、实际内存访问模式(来自编译器代码生成)是不规则的“GATHER”,但从逻辑上讲,您的访问模式是规则的(不变或单位步长)。

解决方案可能不是很容易,但我会尝试以下操作:

  1. 如果算法允许 - 考虑排除 if 语句。这有时可以通过将循环分成几个来实现。
  2. 尝试使用半浮点归纳变量、地板等。尝试使它们成为整数并使用“规范”形式(for (i) array [super-simple-expression(i)] = something)
  3. 尝试使用 pragma simd 的线性子句来通知编译器实际上某处存在单位步长
于 2020-05-21T10:35:15.840 回答