@@ -2464,11 +2464,13 @@ TEST_P(Operator, IntRelu) {
2464
2464
auto *relu = F_->createRELU (" relu" , rescale, reluOutTy);
2465
2465
auto *dequantize = F_->createDequantize (" dequantize" , relu);
2466
2466
2467
- auto *save = F_->createSave (" save" , dequantize);
2467
+ auto *save = F_->createSave (ctx_, " save" , dequantize);
2468
+ ctx_.allocate (mod_.getPlaceholders ());
2469
+
2468
2470
EE_.compile (CompilationMode::Infer, F_, ctx_);
2469
2471
EE_.run ();
2470
2472
2471
- auto result = save->getVariable ( )->getHandle ();
2473
+ auto result = ctx_. get ( save->getPlaceholder () )->getHandle ();
2472
2474
float expectedValue = std::max (0 .0f , splatValue);
2473
2475
for (size_t i = 0 ; i < result.size (); i++) {
2474
2476
EXPECT_EQ (expectedValue, result.raw (i));
@@ -2485,11 +2487,12 @@ TEST_P(Operator, IntSplat) {
2485
2487
auto *splat = F_->createSplat (" splat" , splatTy, splatValue);
2486
2488
auto *dequantize = F_->createDequantize (" dequantize" , splat);
2487
2489
2488
- auto *save = F_->createSave (" save" , dequantize);
2490
+ auto *save = F_->createSave (ctx_, " save" , dequantize);
2491
+ ctx_.allocate (mod_.getPlaceholders ());
2489
2492
EE_.compile (CompilationMode::Infer, F_, ctx_);
2490
2493
EE_.run ();
2491
2494
2492
- auto result = save->getVariable ( )->getHandle ();
2495
+ auto result = ctx_. get ( save->getPlaceholder () )->getHandle ();
2493
2496
for (size_t i = 0 ; i < result.size (); i++) {
2494
2497
EXPECT_EQ (splatValue, result.raw (i));
2495
2498
}
0 commit comments