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 e4ec3aa commit 1836ed0Copy full SHA for 1836ed0
csrc/scheduler/cache_policy_refiner.cpp
@@ -58,6 +58,12 @@ bool isLoadGlobalToLocal(const Expr* expr) {
58
if (ldst->opType() != LoadStoreOpType::Set) {
59
return false;
60
}
61
+ // It should not be necessary to check the output since it should be
62
+ // always a TensorView as long as the input is a TensorView, but
63
+ // just in case.
64
+ if (!ldst->in()->isA<TensorView>() || !ldst->out()->isA<TensorView>()) {
65
+ return false;
66
+ }
67
if (ldst->in()->as<TensorView>()->getMemoryType() != MemoryType::Global) {
68
69
0 commit comments