Skip to content

Commit 8e975da

Browse files
authored
Merge pull request #2722 from ROCm/r2.13-rocm-enhanced-gfx942-hlo-analyse
gcc fix for topk
2 parents 0c0fe34 + 799aab2 commit 8e975da

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tensorflow/compiler/xla/service/topk_rewriter.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@ limitations under the License.
1515

1616
#include "tensorflow/compiler/xla/service/topk_rewriter.h"
1717

18+
#include <array>
1819
#include <memory>
1920
#include <optional>
2021
#include <vector>
@@ -151,7 +152,7 @@ static bool IsNanSafeGt(HloComputation* comp) {
151152
// Look for the instructions emitted from: xla/client/lib/sorting.cc
152153
static bool HasIota(HloSortInstruction* sort, HloInstruction* data) {
153154
namespace m = match;
154-
const auto sort_dims = {data->shape().dimensions(sort->sort_dimension())};
155+
const std::array<int64_t, 1> sort_dims = {data->shape().dimensions(sort->sort_dimension())};
155156
auto match_iota = [](auto dims) {
156157
return m::Iota().WithShape(m::Shape().WithElementType(S32).WithDims(dims));
157158
};

0 commit comments

Comments
 (0)