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 1a22204 commit 56b5a53Copy full SHA for 56b5a53
core/compiler.cpp
@@ -235,8 +235,11 @@ GraphAndMapping ConstructFallbackGraph(
235
236
// the mapping from lowering graph => fallback global graph
237
std::unordered_map<torch::jit::Value*, torch::jit::Value*> old_to_new_g;
238
- for (auto input : block->inputs()) {
239
- util::getOrAddInputForValue(input, new_g, old_to_new_g);
+
+ for (uint64_t i = 0; i < block->inputs().size(); i++) {
240
+ auto in_val = new_g->addInput(std::string("input_") + std::to_string(i));
241
+ in_val->setType(block->inputs()[i]->type());
242
+ old_to_new_g[block->inputs()[i]] = in_val;
243
}
244
245
for (auto& seg_block : segmented_blocks) {
0 commit comments