ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-05 14:11:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

support atomic_pk_add_bf16 on gfx950

Browse Source
This commit is contained in:
Feng Shijie
2025-08-21 09:12:21 +00:00
parent 85976b0b87
commit c378e9bdf8
4 changed files with 97 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -151,6 +151,9 @@ using bf16x16_t = bf16_raw_t __attribute__((ext_vector_type(16)));
using bf16x32_t = bf16_raw_t __attribute__((ext_vector_type(32)));
using bf16x64_t = bf16_raw_t __attribute__((ext_vector_type(64)));
using cktile_llvm_bf16x2_t = __bf16 __attribute__((ext_vector_type(2)));
using cktile_llvm_bf16x4_t = __bf16 __attribute__((ext_vector_type(4)));
// i32
// using int32_t = ...
using int32x2_t = int32_t __attribute__((ext_vector_type(2)));