Skip to content

Commit 78ff3f9

Browse files
committed
[Bugfix] Disable SingleEnvThreadVerifier
During TensorIR scheduling, the `IterVar`s that represent environment threads may duplicate, i.e. it is legal to have two env threads with the same name tag, which may fail the `SingleEnvThreadVerifier` check during schedule creation. This PR disables this check in this case. In the future, it may be worthwhile to bring it back against post-scheduling TIR.
1 parent 0b2358c commit 78ff3f9

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

src/tir/analysis/verify_well_formed.cc

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -347,7 +347,6 @@ bool VerifyWellFormed(const PrimFunc& func, bool assert_mode) {
347347
}
348348

349349
if (!UndefinedVarVerifier::Verify(func, assert_mode)) return false;
350-
if (!SingleEnvThreadVerifier::Verify(func, assert_mode)) return false;
351350

352351
// TODO(Siyuan): add more checks here.
353352
return true;
@@ -364,7 +363,6 @@ bool VerifyWellFormed(const IRModule& mod, bool assert_mode) {
364363
}
365364

366365
if (!UndefinedVarVerifier::Verify(mod, assert_mode)) return false;
367-
if (!SingleEnvThreadVerifier::Verify(mod, assert_mode)) return false;
368366

369367
return true;
370368
}

0 commit comments

Comments
 (0)