Skip to content

Commit 5f43194

Browse files
authored
[AMD] Add gfx1250 skeleton (#8131)
1 parent 9c3563d commit 5f43194

File tree

4 files changed

+6
-1
lines changed

4 files changed

+6
-1
lines changed
Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from ._layouts import AMDMFMALayout, AMDWMMALayout
22
from . import cdna3, cdna4
33
from . import rdna3, rdna4
4+
from . import gfx1250
45

5-
__all__ = ["AMDMFMALayout", "AMDWMMALayout", "cdna3", "cdna4", "rdna3", "rdna4"]
6+
__all__ = ["AMDMFMALayout", "AMDWMMALayout", "cdna3", "cdna4", "rdna3", "rdna4", "gfx1250"]

python/triton/experimental/gluon/language/amd/gfx1250/__init__.py

Whitespace-only changes.

third_party/amd/include/TritonAMDGPUToLLVM/TargetUtils.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ enum class ISAFamily {
1616
RDNA2,
1717
RDNA3,
1818
RDNA4,
19+
GFX1250,
1920
};
2021

2122
// Deduces the corresponding ISA family for the given target gfx |arch|.

third_party/amd/lib/TritonAMDGPUToLLVM/TargetUtils.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,9 @@ ISAFamily deduceISAFamily(llvm::StringRef arch) {
99
// See https://llvm.org/docs/AMDGPUUsage.html#processors for how to categorize
1010
// the following target gfx architectures.
1111

12+
if (kind == llvm::AMDGPU::GK_GFX1250)
13+
return ISAFamily::GFX1250;
14+
1215
// CDNA ISA cases
1316
switch (kind) {
1417
case llvm::AMDGPU::GK_GFX950:

0 commit comments

Comments
 (0)