0

我想使用 SSE3 指令更改以下代码：

 for (i=0; i<=imax+1; i++) {
        /* The vertical velocity approaches 0 at the north and south
         * boundaries, but fluid flows freely in the horizontal direction */
        v[i][jmax] = 0.0;
        u[i][jmax+1] = u[i][jmax];

        v[i][0] = 0.0;
        u[i][0] = u[i][1];
    }

u 和 v 是浮点类型的二维数组。到目前为止我所拥有的是这个，但程序运行不正确。

    int loop2 = ((imax+1) / loopFactor) * loopFactor;
    for(i=0; i<loop2; i+=loopFactor) {
        
        __m128 zeroVec = _mm_set1_ps(0.0f);
        _mm_storeu_ps(&v[i][jmax], zeroVec);
        __m128 umaxVec = _mm_loadu_ps(&u[i][jmax]);
        _mm_storeu_ps(&u[i][jmax+1], umaxVec);

        __m128 zVec = _mm_set1_ps(0.0f);
        _mm_storeu_ps(&v[i][0], zVec);
        __m128 uVec = _mm_loadu_ps(&u[i][1]);
        _mm_storeu_ps(&u[i][0], uVec);
    }
    for (; i<=imax+1; i++){
        v[i][jmax] = 0.0;
        u[i][jmax+1] = u[i][jmax];

        v[i][0] = 0.0;
        u[i][0] = u[i][1];
    }

我怀疑这是因为_mm_loadu_ps存储 u[i][1]、u[i][2]、u[i][3] 和 u[i][4] 的值，但我想存储 u[ i][1]、u[i+1][1]、u[i+2][1]、u[i+3][1] 和 u[i+4][1]。我怎样才能做到这一点？Loopfactor 的值为 4。非常感谢任何帮助。

4

0 回答 0