Skip to content

Commit f3f1a92

Browse files
committed
[Placeholder] Update two tests that were creating variables that were saved into.
1 parent 2c16dc9 commit f3f1a92

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

tests/unittests/OperatorTest.cpp

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2464,11 +2464,13 @@ TEST_P(Operator, IntRelu) {
24642464
auto *relu = F_->createRELU("relu", rescale, reluOutTy);
24652465
auto *dequantize = F_->createDequantize("dequantize", relu);
24662466

2467-
auto *save = F_->createSave("save", dequantize);
2467+
auto *save = F_->createSave(ctx_, "save", dequantize);
2468+
ctx_.allocate(mod_.getPlaceholders());
2469+
24682470
EE_.compile(CompilationMode::Infer, F_, ctx_);
24692471
EE_.run();
24702472

2471-
auto result = save->getVariable()->getHandle();
2473+
auto result = ctx_.get(save->getPlaceholder())->getHandle();
24722474
float expectedValue = std::max(0.0f, splatValue);
24732475
for (size_t i = 0; i < result.size(); i++) {
24742476
EXPECT_EQ(expectedValue, result.raw(i));
@@ -2485,11 +2487,12 @@ TEST_P(Operator, IntSplat) {
24852487
auto *splat = F_->createSplat("splat", splatTy, splatValue);
24862488
auto *dequantize = F_->createDequantize("dequantize", splat);
24872489

2488-
auto *save = F_->createSave("save", dequantize);
2490+
auto *save = F_->createSave(ctx_, "save", dequantize);
2491+
ctx_.allocate(mod_.getPlaceholders());
24892492
EE_.compile(CompilationMode::Infer, F_, ctx_);
24902493
EE_.run();
24912494

2492-
auto result = save->getVariable()->getHandle();
2495+
auto result = ctx_.get(save->getPlaceholder())->getHandle();
24932496
for (size_t i = 0; i < result.size(); i++) {
24942497
EXPECT_EQ(splatValue, result.raw(i));
24952498
}

0 commit comments

Comments
 (0)