Skip to content

Commit 13c3de1

Browse files
authored
Restore RocmComputeCapability:: gfx11_rx7900() and gfx12_rx8900() methods (#333)
At least gfx11_rx7900() is still needed for TF build.
1 parent 51a7f4b commit 13c3de1

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

xla/stream_executor/device_description.h

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -138,11 +138,20 @@ class RocmComputeCapability {
138138
"gfx1151"};
139139
bool gfx11_apu() const { return IsThisGfxInAnyList(kGfx11Apu); }
140140

141+
static constexpr absl::string_view kGfx11Rx7900[] = {"gfx1100", "gfx1101",
142+
"gfx1102"};
143+
bool gfx11_rx7900() const {
144+
// TODO(AMD/TF): instead of this, other gfx11*() methods might be better
145+
return IsThisGfxInAnyList(kGfx11Rx7900);
146+
}
147+
141148
bool gfx12() const { return absl::StartsWith(gfx_version(), "gfx12"); }
142149

143150
static constexpr absl::string_view kGfx12Discrete[] = {"gfx1200", "gfx1201"};
144151
bool gfx12_discrete() const { return IsThisGfxInAnyList(kGfx12Discrete); }
145152

153+
bool gfx12_rx8900() const { return gfx12_discrete(); }
154+
146155
bool has_nhwc_layout_support() const { return gfx9_mi100_or_later(); }
147156

148157
bool has_bf16_dtype_support() const {

0 commit comments

Comments
 (0)