Skip to content

Commit 0eb9d7e

Browse files
wsmosesGoogle-ML-Automation
authored andcommitted
TileAssignment: Fix build with gcc
PiperOrigin-RevId: 873406125
1 parent 964a0a4 commit 0eb9d7e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

xla/hlo/utils/hlo_sharding_util.h

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -372,7 +372,8 @@ class DeviceGroupTileAssignment : public TileAssignment {
372372
public:
373373
explicit DeviceGroupTileAssignment(int64_t num_groups,
374374
int64_t num_devices_per_group)
375-
: TileAssignment({num_groups, num_devices_per_group}) {}
375+
: TileAssignment(
376+
absl::Span<const int64_t>{num_groups, num_devices_per_group}) {}
376377
explicit DeviceGroupTileAssignment(int64_t num_groups,
377378
int64_t num_devices_per_group,
378379
absl::Span<const int64_t> reshape_dims,

0 commit comments

Comments
 (0)