From 11e4082dd8459f3a0e69f7d164ef64eb7ebfa7fa Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Bart=C5=82omiej=20Kocot?= Date: Tue, 4 Feb 2025 14:42:11 +0100 Subject: [PATCH] Fix duplication of pk_add_f16 symbols (#1858) --- include/ck_tile/core/numeric/vector_type.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/include/ck_tile/core/numeric/vector_type.hpp b/include/ck_tile/core/numeric/vector_type.hpp index 9aeb494919..480da96596 100644 --- a/include/ck_tile/core/numeric/vector_type.hpp +++ b/include/ck_tile/core/numeric/vector_type.hpp @@ -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));