ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-19 20:40:07 +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/36_sparse_embedding/sparse_embedding3_forward_layernorm.cpp
View File

@@ -19,6 +19,10 @@
#include "ck/library/utility/host_tensor_generator.hpp"
#include "ck/library/reference_tensor_operation/cpu/reference_sparse_embedding3_forward_layernorm.hpp"
using ::ck::DeviceMem;
using ::ck::HostTensorDescriptor;
using ::ck::Tensor;
// clang-format off
using EmbType = ck::half_t;
using IndexType = int64_t;