diff options
author | Koichi Akabe <vbkaisetsu@gmail.com> | 2018-10-30 14:54:55 +0900 |
---|---|---|
committer | Koichi Akabe <vbkaisetsu@gmail.com> | 2018-10-30 14:54:55 +0900 |
commit | 0b3d04f70902e00f86c572a5e3c379f9335b216f (patch) | |
tree | fdc57080f0ea3da59699907129cbf13f1374cea0 /src/utilities | |
parent | d45911b61dedafcbd74f65df263b4197697d6a81 (diff) |
Fix col2im implementation
Diffstat (limited to 'src/utilities')
-rw-r--r-- | src/utilities/utilities.cpp | 26 | ||||
-rw-r--r-- | src/utilities/utilities.hpp | 6 |
2 files changed, 32 insertions, 0 deletions
diff --git a/src/utilities/utilities.cpp b/src/utilities/utilities.cpp index a8fdaa19..a6cd82e7 100644 --- a/src/utilities/utilities.cpp +++ b/src/utilities/utilities.cpp @@ -489,4 +489,30 @@ std::string GetDeviceName(const Device& device) { } // ================================================================================================= + +// Solve Bezout's identity +// a * p + b * q = r = GCD(a, b) +void EuclidGCD(int a, int b, int &p, int &q, int &r) { + p = 0; + q = 1; + int p_1 = 1; + int q_1 = 0; + for (;;) { + const int c = a % b; + if (c == 0) { + break; + } + const int p_2 = p_1; + const int q_2 = q_1; + p_1 = p; + q_1 = q; + p = p_2 - p_1 * (a / b); + q = q_2 - q_1 * (a / b); + a = b; + b = c; + } + r = b; +} + +// ================================================================================================= } // namespace clblast diff --git a/src/utilities/utilities.hpp b/src/utilities/utilities.hpp index 16a241af..fcc1c57f 100644 --- a/src/utilities/utilities.hpp +++ b/src/utilities/utilities.hpp @@ -372,6 +372,12 @@ std::string GetDeviceArchitecture(const Device& device); std::string GetDeviceName(const Device& device); // ================================================================================================= + +// Solve Bezout's identity +// a * p + b * q = r = GCD(a, b) +void EuclidGCD(int a, int b, int &p, int &q, int &r); + +// ================================================================================================= } // namespace clblast // CLBLAST_UTILITIES_H_ |