File tree Expand file tree Collapse file tree 1 file changed +9
-0
lines changed Expand file tree Collapse file tree 1 file changed +9
-0
lines changed Original file line number Diff line number Diff 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 {
You can’t perform that action at this time.
0 commit comments