Skip to content

Commit 5ba9343

Browse files
authored
Fixes #1523 (#1525)
1 parent 9522b1b commit 5ba9343

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

torch/csrc/jit/codegen/cuda/lower_validation.cpp

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -431,7 +431,10 @@ class VectorizeValidator : public OptInDispatch {
431431
vector_size,
432432
" however, vector sizes only upto and including 16 bytes are supported.");
433433

434-
auto replay_exprs = StmtSort::getExprs(fusion, {v_id}, false);
434+
auto replay_exprs = DependencyCheck::getAllExprsBetween(
435+
{tv->getMaybeRFactorDomain().begin(),
436+
tv->getMaybeRFactorDomain().end()},
437+
{v_id});
435438

436439
VectorizeValidator validator(v_id);
437440

0 commit comments

Comments
 (0)