We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0257c4c commit c07886cCopy full SHA for c07886c
third_party/amd/lib/TritonAMDGPUToLLVM/TargetInfo.cpp
@@ -81,8 +81,15 @@ int TargetInfo::getWarpSize() const {
81
}
82
83
int TargetInfo::getSharedMemorySize() const {
84
- int kbytes = getISAFamily() == ISAFamily::CDNA4 ? 160 : 64;
85
- return kbytes * 1024;
+ // Should return the maximum capacity in kbyte
+ 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
+ }
93
94
95
bool TargetInfo::supportMaximumMinimum() const {
0 commit comments