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

added implicit gemm v1r3 lds_double_buffer NCHW * CYXK = KNHW, reworked static functionals

Browse Source
This commit is contained in:
Chao Liu
2019-04-23 17:51:14 -05:00
parent 87d8740bf5
commit 569ad66e2a
22 changed files with 2117 additions and 1107 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/include/Array.hip.hpp
View File

@@ -19,6 +19,20 @@ struct Array
__host__ __device__ const TData& operator[](index_t i) const { return mData[i]; }
__host__ __device__ TData& operator[](index_t i) { return mData[i]; }
__host__ __device__ auto PushBack(TData x) const
{
Array<TData, NSize + 1> new_array;
static_for<0, NSize, 1>{}([=](auto I) {
constexpr index_t i = I.Get();
new_array[i] = mData[i];
});
new_array[NSize] = x;
return new_array;
}
};
template <class TData, index_t NSize, index_t... IRs>
@@ -51,4 +65,4 @@ __host__ __device__ auto reorder_array_given_old2new(const Array<TData, NSize>&
});
return new_array;
}
}