Skip to content

Commit c07886c

Browse files
authored
[AMD] Update shared memory size for gfx1250 from TargetInfo (#8517)
Add shared memory capacity for `gfx1250` which is 320 kbyte.
1 parent 0257c4c commit c07886c

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

third_party/amd/lib/TritonAMDGPUToLLVM/TargetInfo.cpp

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -81,8 +81,15 @@ int TargetInfo::getWarpSize() const {
8181
}
8282

8383
int TargetInfo::getSharedMemorySize() const {
84-
int kbytes = getISAFamily() == ISAFamily::CDNA4 ? 160 : 64;
85-
return kbytes * 1024;
84+
// Should return the maximum capacity in kbyte
85+
switch (getISAFamily()) {
86+
case ISAFamily::GFX1250:
87+
return 320 * 1024;
88+
case ISAFamily::CDNA4:
89+
return 160 * 1024;
90+
default:
91+
return 64 * 1024;
92+
}
8693
}
8794

8895
bool TargetInfo::supportMaximumMinimum() const {

0 commit comments

Comments
 (0)