Skip to content

Commit 0e6c80d

Browse files
jakeh-gcgeorgepaw
authored andcommitted
Request tile pairs.
Summary: Fixes T43234 Depends on D48974 Reviewers: #tensorflow, simonl, #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved, georgep Reviewed By: #tensorflow, #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved, georgep Maniphest Tasks: T43234 Differential Revision: https://phabricator.sourcevertex.net/D48975
1 parent 3b02127 commit 0e6c80d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tensorflow/compiler/plugin/poplar/driver/poplar_compiler.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -678,7 +678,8 @@ absl::optional<Tilesets> PartitionTiles(const poplar::Graph& main_graph,
678678
const auto num_compute_tiles = num_tiles_per_ipu - num_io_tiles;
679679

680680
const auto compute_tiles =
681-
gcl::perIPUTiles(main_graph, num_io_tiles, num_compute_tiles);
681+
gcl::perIPUTiles(main_graph, num_io_tiles, num_compute_tiles,
682+
/*sorted=*/true, /*tilePairs=*/true);
682683
CHECK_EQ(compute_tiles.size(), num_compute_tiles);
683684

684685
const auto io_tiles = DisjointTiles(compute_tiles, num_tiles_per_ipu);

0 commit comments

Comments
 (0)