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 1c11c36 commit 6f3ec49Copy full SHA for 6f3ec49
core/partitioning/partitioning.cpp
@@ -74,7 +74,7 @@ void setExplicitFallbackNodes(PartitioningCtx* ctx, torch::jit::Block* block) {
74
continue;
75
}
76
77
- if (n->kind() == torch::jit::prim::Loop && checkLoopEvaluatable) {
+ if (n->kind() == torch::jit::prim::Loop && checkLoopEvaluatable(n)) {
78
ctx->setNodeExecutorDecision(n, NodeExecutorDecision::kCONVERT);
79
} else if (!conversion::OpSupported(n)) {
80
// If the op is not supported by the conversion phase it should run in PyTorch
0 commit comments