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

rename files, added header guard, added namespace

Browse Source 
[ROCm/composable_kernel commit: 88b77181aa]
This commit is contained in:
Chao Liu
2019-06-11 14:35:59 -05:00
parent 88d7d31a76
commit 3dc2ff612c
62 changed files with 580 additions and 1844 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/include/threadwise_generic_tensor_op.hpp Normal file
View File

@@ -0,0 +1,19 @@
#ifndef CK_THREADWISE_GENERIC_TENSOR_OP_HPP
#define CK_THREADWISE_GENERIC_TENSOR_OP_HPP
#include "ConstantTensorDescriptor.hpp"
#include "ConstantMergedTensorDescriptor.hpp"
namespace ck {
template <class Float, class TDesc>
__device__ void threadwise_generic_tensor_set_zero(TDesc, Float* __restrict__ p)
{
static_ford<decltype(TDesc::GetLengths())>{}([&](auto multi_id) {
constexpr index_t offset = TDesc::GetOffsetFromMultiIndex(multi_id);
p[offset] = static_cast<Float>(0);
});
}
} // namespace ck
#endif