Skip to content

Commit e380caf

Browse files
rm numIOTiles <= 192 limit
Summary: !ci_branch_mk2 For some models with a large number of inputs, current max `numIOTiles` is 192, which may cause OOM, this diff removed this limitation Reviewers: #popart, #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved, shirazb Reviewed By: #popart, #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved, shirazb Subscribers: shirazb, leiw, grahamh, matthewha, xiaobingw Maniphest Tasks: T74718 Differential Revision: https://phabricator.sourcevertex.net/D85748
1 parent f565723 commit e380caf

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

willow/src/popx/irlowering.cpp

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2927,11 +2927,10 @@ void IrLowering::prepareGraph() {
29272927

29282928
if (numIOTiles > 0) {
29292929

2930-
if (numIOTiles < 32 || numIOTiles > 192 || (numIOTiles % 2 != 0)) {
2931-
throw error(
2932-
"{} is an invalid number of IO tiles. "
2933-
"Number of IO tiles must be an even number in range [32, 192]",
2934-
numIOTiles);
2930+
if (numIOTiles < 32 || (numIOTiles % 2 != 0)) {
2931+
throw error("{} is an invalid number of IO tiles. Number of IO tiles "
2932+
"must be an even number and >= 32.",
2933+
numIOTiles);
29352934
}
29362935
logging::devicex::info(
29372936
"Reserving {} IO tiles for GCL collective operations on each IPU",

0 commit comments

Comments
 (0)