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

Fix duplication of pk_add_f16 symbols (#1858)

Browse Source
This commit is contained in:
Bartłomiej Kocot
2025-02-04 14:42:11 +01:00
committed by GitHub
parent 9ee69dd297
commit 11e4082dd8
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
include/ck_tile/core/numeric/vector_type.hpp
View File

@@ -200,7 +200,7 @@ using bf8x32_t = bf8_t __attribute((ext_vector_type(32)));
using bf8x64_t = bf8_t __attribute((ext_vector_type(64)));
#endif
__host__ fp16x2_t pk_add_f16(const fp16x2_t& x, const fp16x2_t& y)
CK_TILE_HOST fp16x2_t pk_add_f16(const fp16x2_t& x, const fp16x2_t& y)
{
fp16x2_t vector_res;
@@ -210,7 +210,7 @@ __host__ fp16x2_t pk_add_f16(const fp16x2_t& x, const fp16x2_t& y)
return vector_res;
}
__device__ fp16x2_t pk_add_f16(const fp16x2_t& x, const fp16x2_t& y)
CK_TILE_DEVICE fp16x2_t pk_add_f16(const fp16x2_t& x, const fp16x2_t& y)
{
fp16x2_t c;
asm volatile("v_pk_add_f16 %0, %1, %2" : "=v"(c) : "v"(x), "v"(y));