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

Merge commit 'ad57f6ef0bcaeef7988bfd3954aac06554f12afb' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-11-19 11:12:09 +00:00
parent 05f83b643f
commit 055f643f4f
235 changed files with 964 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
example/45_elementwise_normalization/elementwise_layernorm_blockwise.cpp
View File

@@ -19,6 +19,10 @@
#include "ck/library/utility/host_tensor_generator.hpp"
#include "ck/library/reference_tensor_operation/cpu/reference_layernorm.hpp"
using ::ck::DeviceMem;
using ::ck::HostTensorDescriptor;
using ::ck::Tensor;
using ADataType = ck::half_t; // Input 1
using BDataType = ck::half_t; // Input 2
using XDataType = ck::half_t;