diff --git a/tests/unittests/GraphOptzTest.cpp b/tests/unittests/GraphOptzTest.cpp index 205c6f5598..740141e2c3 100644 --- a/tests/unittests/GraphOptzTest.cpp +++ b/tests/unittests/GraphOptzTest.cpp @@ -3007,9 +3007,14 @@ TEST_F(GraphOptz, nopRelu) { auto *relu = F_->createRELU("relu", in); F_->createSave("save", relu); - ::glow::optimize(F_, CompilationMode::Infer); + optimizedF_ = optimizeFunction(F_); - EXPECT_EQ(F_->getNodes().size(), 1); + EXPECT_EQ(optimizedF_->getNodes().size(), 1); + + bindings_.allocate(mod_.getPlaceholders()); + bindings_.get(in)->getHandle().randomize(-4, 4, mod_.getPRNG()); + + checkNumericalEquivalence(); } template