Skip to content

Commit c4472d5

Browse files
committed
feat(ck): add gfx90c build support (cherry-pick rocm-libraries#9333)
1 parent 5e18469 commit c4472d5

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

‎include/ck/ck.hpp‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,8 @@
7979
// buffer resource
8080
#ifndef __HIP_DEVICE_COMPILE__ // for host code
8181
#define CK_BUFFER_RESOURCE_3RD_DWORD -1
82-
#elif defined(__gfx803__) || defined(__gfx900__) || defined(__gfx906__) || defined(__gfx9__)
82+
#elif defined(__gfx803__) || defined(__gfx900__) || defined(__gfx90c__) || defined(__gfx906__) || \
83+
defined(__gfx9__)
8384
#define CK_BUFFER_RESOURCE_3RD_DWORD 0x00020000
8485
#elif defined(__gfx101__) || defined(__gfx103__)
8586
#define CK_BUFFER_RESOURCE_3RD_DWORD 0x31014000
@@ -94,7 +95,7 @@
9495
#define CK_USE_AMD_V_FMAC_F32
9596
#define CK_USE_AMD_V_DOT2_F32_F16
9697
#define CK_USE_AMD_V_DOT4_I32_I8
97-
#elif defined(__gfx803__) || defined(__gfx900__) || defined(__gfx101__)
98+
#elif defined(__gfx803__) || defined(__gfx900__) || defined(__gfx90c__) || defined(__gfx101__)
9899
#define CK_USE_AMD_V_MAC_F32
99100
#elif defined(__gfx11__) || defined(__gfx12__)
100101
#define CK_USE_AMD_V_FMAC_F32

0 commit comments

Comments
 (0)