Fix uses of MaxPool without NodeValue (#3232)

Summary: This was broken as a result of #3146 Pull Request resolved: #3232 Test Plan: Ran mnist and it works again 🙂 Reported in [PyTorch forums here](https://discuss.pytorch.org/t/glow-example-run-error/50461). Differential Revision: D16232983 Pulled By: jfix71 fbshipit-source-id: f0d636c0a01a4bb1f673b1d1f47f4392a468da6d
pytorch · Jul 13, 2019 · 4a099d7 · 4a099d7
1 parent a951d72
commit 4a099d7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/mnist.cpp b/examples/mnist.cpp
@@ -95,11 +95,11 @@ void createModel(ExecutionEngine &EE, Function *F,
   auto *RL0 = F->createRELU("relu", CV0);
   auto *MP0 = F->createMaxPool("pool", RL0, 3, 3, 0);
 
-  auto *CV1 = F->createConv(bindings, "conv", MP0, 16, 5, 1, 2, 1);
+  auto *CV1 = F->createConv(bindings, "conv", MP0->getResult(), 16, 5, 1, 2, 1);
   auto *RL1 = F->createRELU("relu", CV1);
   auto *MP1 = F->createMaxPool("pool", RL1, 3, 3, 0);
 
-  auto *FCL1 = F->createFullyConnected(bindings, "fc", MP1, 10);
+  auto *FCL1 = F->createFullyConnected(bindings, "fc", MP1->getResult(), 10);
   selectedPH = mod.createPlaceholder(ElemKind::Int64ITy, {minibatchSize, 1},
                                      "selected", false);
   auto *SM = F->createSoftMax("sm", FCL1, selectedPH);