ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-19 12:30:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

MIopen integration (#13)

Browse Source 
* update for miopen integration: cosmetic refactor


[ROCm/composable_kernel commit: 1a66e35b6f]
This commit is contained in:
Chao Liu
2020-02-17 09:53:20 -06:00
committed by GitHub
parent 81e3c745dc
commit ef393a2bb2
29 changed files with 764 additions and 647 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
composable_kernel/include/utility/math.hpp
View File

@@ -107,27 +107,22 @@ __host__ __device__ constexpr T min(T x, Ts... xs)
template <typename T>
__host__ __device__ constexpr T gcd(T x, T y)
{
if(x == 0)
if(x == y || x == 0)
{
return y;
}
if(y == 0)
else if(y == 0)
{
return x;
}
if(x == y)
{
return x;
}
if(x > y)
else if(x > y)
{
return gcd(x - y, y);
}
return gcd(x, y - x);
else
{
return gcd(x, y - x);
}
}
template <index_t X, index_t Y>
@@ -150,10 +145,10 @@ __host__ __device__ constexpr T lcm(T x, T y)
return (x * y) / gcd(x, y);
}
template <typename X, typename Y, typename... Zs>
__host__ __device__ constexpr auto lcm(X x, Y y, Zs... zs)
template <typename X, typename... Ys>
__host__ __device__ constexpr auto lcm(X x, Ys... ys)
{
return lcm(x, lcm(y, zs...));
return lcm(x, lcm(ys...));
}
template <class T>