瀏覽代碼

solved scaling fma problem

Nicolas Winkler 5 年之前
父節點
當前提交
3ecb70294a
共有 3 個文件被更改,包括 12 次插入182 次删除
  1. 4 4
      MandelWidget.cpp
  2. 8 1
      libmandel/CMakeLists.txt
  3. 0 177
      libmandel/src/CpuGeneratorsAVX.cpp

+ 4 - 4
MandelWidget.cpp

@@ -541,10 +541,10 @@ void MandelView::paint(const mnd::MandelViewport& mvp)
             if (t != nullptr) {
                 t->img->drawRect(float(x), float(y), float(w), float(w));
                 /*glBegin(GL_LINE_LOOP);
-                glVertex2f(x, y);
-                glVertex2f(x + w, y);
-                glVertex2f(x + w, y + w);
-                glVertex2f(x, y + w);
+                glVertex2f(float(x), float(y));
+                glVertex2f(float(x) + float(w), float(y));
+                glVertex2f(float(x) + float(w), float(y) + float(w));
+                glVertex2f(float(x), float(y) + float(w));
                 glEnd();*/
 
                 if (!t->enoughResolution) {

+ 8 - 1
libmandel/CMakeLists.txt

@@ -38,7 +38,7 @@ SET(MandelSources
 FILE(GLOB MandelHeaders include/*.h)
 
 if (ARCH STREQUAL "X86_64" OR ARCH STREQUAL "X86")
-    list(APPEND MandelSources src/CpuGeneratorsAVX.cpp src/CpuGeneratorsSSE2.cpp)
+    list(APPEND MandelSources src/CpuGeneratorsAVX.cpp src/CpuGeneratorsAVXFMA.cpp src/CpuGeneratorsSSE2.cpp)
 elseif(ARCH STREQUAL "ARM")
     list(APPEND MandelSources src/CpuGeneratorsNeon.cpp)
 endif()
@@ -96,10 +96,17 @@ endif(Boost_FOUND)
 if (ARCH STREQUAL "X86_64" OR ARCH STREQUAL "X86")
     if (MSVC)
         set_source_files_properties(src/CpuGeneratorsAVX.cpp PROPERTIES COMPILE_FLAGS /arch:AVX)
+        set_source_files_properties(src/CpuGeneratorsAVXFMA.cpp PROPERTIES COMPILE_FLAGS /arch:AVX)
     else()
         set_source_files_properties(src/CpuGeneratorsAVX.cpp PROPERTIES COMPILE_FLAGS -mavx)
+        set_source_files_properties(src/CpuGeneratorsAVXFMA.cpp PROPERTIES COMPILE_FLAGS -mavx)
     endif(MSVC)
 
+    if (MSVC)
+        set_source_files_properties(src/CpuGeneratorsAVXFMA.cpp PROPERTIES COMPILE_FLAGS /arch:FMA)
+    else()
+        set_source_files_properties(src/CpuGeneratorsAVXFMA.cpp PROPERTIES COMPILE_FLAGS -mfma)
+    endif(MSVC)
 
     if (MSVC)
         set_source_files_properties(src/CpuGeneratorsSSE2.cpp PROPERTIES COMPILE_FLAGS /arch:SSE2)

+ 0 - 177
libmandel/src/CpuGeneratorsAVX.cpp

@@ -16,9 +16,6 @@ namespace mnd
 
     template class CpuGenerator<double, mnd::X86_AVX, false>;
     template class CpuGenerator<double, mnd::X86_AVX, true>;
-
-    template class CpuGenerator<DoubleDouble, mnd::X86_AVX_FMA, false>;
-    template class CpuGenerator<DoubleDouble, mnd::X86_AVX_FMA, true>;
 }
 
 template<bool parallel>
@@ -166,177 +163,3 @@ void CpuGenerator<double, mnd::X86_AVX, parallel>::generate(const mnd::MandelInf
     }
 }
 
-static inline std::pair<__m256d, __m256d> quickTwoSum(__m256d a, __m256d b)
-{
-    __m256d s = _mm256_add_pd(a, b);
-    __m256d e = _mm256_sub_pd(b, _mm256_sub_pd(s, a));
-    return { s, e };
-}
-
-static inline std::pair<__m256d, __m256d> quickTwoDiff(__m256d a, __m256d b)
-{
-    __m256d s = _mm256_sub_pd(a, b);
-    __m256d e = _mm256_sub_pd(_mm256_sub_pd(a, s), b);
-    return { s, e };
-}
-
-static inline std::pair<__m256d, __m256d> twoSum(__m256d a, __m256d b)
-{
-    __m256d s = _mm256_add_pd(a, b);
-    __m256d bb = _mm256_sub_pd(s, a);
-    __m256d e = _mm256_add_pd(_mm256_sub_pd(a, _mm256_sub_pd(s, bb)), _mm256_sub_pd(b, bb));
-    return { s, e };
-}
-
-static inline std::pair<__m256d, __m256d> twoDiff(__m256d a, __m256d b)
-{
-    __m256d s = _mm256_sub_pd(a, b);
-    __m256d bb = _mm256_sub_pd(s, a);
-    __m256d e = _mm256_sub_pd(_mm256_sub_pd(a, _mm256_sub_pd(s, bb)), _mm256_add_pd(b, bb));
-    return { s, e };
-}
-
-/*
-static inline std::pair<__m256d, __m256d> split(__m256d a)
-{
-    static const __m256d SPLIT_THRESH = { 6.69692879491417e+299, 6.69692879491417e+299, 6.69692879491417e+299, 6.69692879491417e+299 };
-    static const __m256d MINUS_SPLIT_THRESH = { -6.69692879491417e+299, -6.69692879491417e+299, -6.69692879491417e+299, -6.69692879491417e+299 };
-    static const __m256d SPLITTER = { 134217729.0, 134217729.0, 134217729.0, 134217729.0};
-    __m256d temp;
-    __m256i cmp1 = _mm256_castpd_si256(_mm256_cmp_pd(a, SPLIT_THRESH, _CMP_GT_OQ));
-    __m256i cmp2 = _mm256_castpd_si256(_mm256_cmp_pd(a, MINUS_SPLIT_THRESH, _CMP_LT_OQ));
-    __m256i cmp = _mm256_or_si256
-}*/
-
-static inline std::pair<__m256d, __m256d> twoProd(__m256d a, __m256d b)
-{
-//#ifdef CPUID_FMA
-    __m256d p = _mm256_mul_pd(a, b);
-    __m256d e = _mm256_fmadd_pd(a, b, p);
-    return { p, e };
-//#else
-/*    double a_hi, a_lo, b_hi, b_lo;
-    __m256d p = _mm256_mul_ps(a, b);
-    split(a, a_hi, a_lo);
-    split(b, b_hi, b_lo);
-    err = ((a_hi * b_hi - p) + a_hi * b_lo + a_lo * b_hi) + a_lo * b_lo;
-    return p;*/
-//#endif
-}
-
-struct AvxDoubleDouble
-{
-    __m256d x[2];
-
-    inline AvxDoubleDouble(__m256d a, __m256d b) :
-        x{ a, b }
-    {}
-
-
-    inline AvxDoubleDouble operator + (const AvxDoubleDouble& sm) const
-    {
-        auto[s, e] = twoSum(x[0], sm.x[0]);
-        e = _mm256_add_pd(e, _mm256_add_pd(x[1], sm.x[1]));
-        auto[r1, r2] = quickTwoSum(s, e);
-        return AvxDoubleDouble{ r1, r2 };
-    }
-
-    inline AvxDoubleDouble operator - (const AvxDoubleDouble& sm) const
-    {
-        auto[s, e] = twoDiff(x[0], sm.x[0]);
-        e = _mm256_add_pd(e, x[1]);
-        e = _mm256_sub_pd(e, sm.x[1]);
-        auto[r1, r2] = quickTwoSum(s, e);
-        return AvxDoubleDouble{ r1, r2 };
-    }
-
-    inline AvxDoubleDouble operator * (const AvxDoubleDouble& sm) const
-    {
-        auto[p1, p2] = twoProd(this->x[0], sm.x[0]);
-        p2 = _mm256_add_pd(p2,
-            _mm256_add_pd(_mm256_mul_pd(x[0], sm.x[1]), _mm256_mul_pd(x[1], sm.x[0])) );
-        auto[r1, r2] = quickTwoSum(p1, p2);
-        return AvxDoubleDouble{ r1, r2 };
-    }
-};
-
-template<bool parallel>
-void CpuGenerator<mnd::DoubleDouble, mnd::X86_AVX_FMA, parallel>::generate(const mnd::MandelInfo& info, float* data)
-{
-    const MandelViewport& view = info.view;
-
-    using T = DoubleDouble;
-
-    T viewx = mnd::convert<T>(view.x);
-    T viewy = mnd::convert<T>(view.y);
-    T wpp = mnd::convert<T>(view.width / info.bWidth);
-    T hpp = mnd::convert<T>(view.height / info.bHeight);
-
-    if constexpr(parallel)
-        omp_set_num_threads(2 * omp_get_num_procs());
-#pragma omp parallel for schedule(static, 1) if (parallel)
-    for (long j = 0; j < info.bHeight; j++) {
-        T y = viewy + T(double(j)) * hpp;
-        __m256d y0s = { y.x[0], y.x[0], y.x[0], y.x[0] };
-        __m256d y1s = { y.x[1], y.x[1], y.x[1], y.x[1] };
-        AvxDoubleDouble ys{ y0s, y1s };
-        long i = 0;
-        for (i; i < info.bWidth; i += 4) {
-            T x1 = viewx + T(double(i)) * wpp;
-            T x2 = viewx + T(double(i + 1)) * wpp;
-            T x3 = viewx + T(double(i + 2)) * wpp;
-            T x4 = viewx + T(double(i + 3)) * wpp;
-
-            __m256d x0s = {
-                x1.x[0], x2.x[0], x3.x[0], x4.x[0],
-            };
-
-            __m256d x1s = {
-                x1.x[1], x2.x[1], x3.x[1], x4.x[1],
-            };
-
-            AvxDoubleDouble xs{ x0s, x1s };
-
-            int itRes[4] = { 0, 0, 0, 0 };
-
-            __m256d threshold = { 16.0, 16.0, 16.0, 16.0 };
-            __m256d counter = { 0, 0, 0, 0 };
-            __m256d adder = { 1, 1, 1, 1 };
-
-            AvxDoubleDouble a = xs;
-            AvxDoubleDouble b = ys;
-
-            for (int k = 0; k < info.maxIter; k++) {
-                AvxDoubleDouble aa = a * a;
-                AvxDoubleDouble bb = b * b;
-                AvxDoubleDouble abab = a * b; abab = abab + abab;
-                a = aa - bb + xs;
-                b = abab + ys;
-                __m256i cmp = _mm256_castpd_si256(_mm256_cmp_pd(_mm256_add_pd(aa.x[0], bb.x[0]), threshold, _CMP_LE_OQ));
-                /*if (info.smooth) {
-                    resultsa = _mm256_or_pd(_mm256_andnot_ps(cmp, resultsa), _mm256_and_ps(cmp, a));
-                    resultsb = _mm256_or_ps(_mm256_andnot_ps(cmp, resultsb), _mm256_and_ps(cmp, b));
-                }*/
-                adder = _mm256_and_pd(adder, _mm256_castsi256_pd(cmp));
-                counter = _mm256_add_pd(counter, adder);
-                if ((k & 0x7) == 0 && _mm256_testz_si256(cmp, cmp) != 0) {
-                    break;
-                }
-            }
-
-            auto alignVec = [](double* data) -> double* {
-                void* aligned = data;
-                ::size_t length = 64;
-                std::align(32, 4 * sizeof(double), aligned, length);
-                return static_cast<double*>(aligned);
-            };
-
-            double resData[8];
-            double* ftRes = alignVec(resData);
-            _mm256_store_pd(ftRes, counter);
-            for (int k = 0; k < 4 && i + k < info.bWidth; k++)
-                data[i + k + j * info.bWidth] = ftRes[k] > 0 ? float(ftRes[k]) : info.maxIter;
-        }
-    }
-}
-