We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9522b1b commit 5ba9343Copy full SHA for 5ba9343
torch/csrc/jit/codegen/cuda/lower_validation.cpp
@@ -431,7 +431,10 @@ class VectorizeValidator : public OptInDispatch {
431
vector_size,
432
" however, vector sizes only upto and including 16 bytes are supported.");
433
434
- auto replay_exprs = StmtSort::getExprs(fusion, {v_id}, false);
+ auto replay_exprs = DependencyCheck::getAllExprsBetween(
435
+ {tv->getMaybeRFactorDomain().begin(),
436
+ tv->getMaybeRFactorDomain().end()},
437
+ {v_id});
438
439
VectorizeValidator validator(v_id);
440
0 commit comments