Skip to content

Commit 15ac7c4

Browse files
authored
tree_node_3D: use diagonal sbrc for gfx1030
1 parent f21724a commit 15ac7c4

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

library/src/include/tree_node_3D.h

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -181,9 +181,11 @@ class SBRCTranspose3DNode : public LeafNode
181181
auto alignment_dimension = sbrc_3D_alignment_dimension();
182182
if(alignment_dimension == 0)
183183
return NONE;
184-
// NB: from the benchmark results, diagonal transpose benefits only gfx906
184+
// NB: from the benchmark results, diagonal transpose
185+
// benefits only some architectures
185186
if(is_diagonal_sbrc_3D_length(length.front()) && is_cube_size(length)
186-
&& is_device_gcn_arch(deviceProp, "gfx906"))
187+
&& (is_device_gcn_arch(deviceProp, "gfx906")
188+
|| is_device_gcn_arch(deviceProp, "gfx1030")))
187189
return DIAGONAL;
188190
if(alignment_dimension % blockWidth == 0)
189191
return TILE_ALIGNED;

0 commit comments

Comments
 (0)