ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-24 06:44:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactoring ConstantTensorDescriptor

Browse Source
This commit is contained in:
Chao Liu
2019-03-17 03:22:41 -05:00
parent fd8de38417
commit a0584426ff
9 changed files with 452 additions and 340 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/include/Array.hip.hpp Normal file
View File

@@ -0,0 +1,18 @@
#pragma once
template <class TData, unsigned NSize>
struct Array
{
using Type = Array<TData, NSize>;
static constexpr unsigned nSize = NSize;
unsigned mData[nSize];
template <class... Xs>
__host__ __device__ Array(Xs... xs) : mData({static_cast<TData>(xs)...})
{
}
__host__ __device__ TData operator[](unsigned i) const { return mData[i]; }
};