diff --git a/test/TestDestroy.cpp b/test/TestDestroy.cpp index 1e87f25b3..01a0e6c78 100644 --- a/test/TestDestroy.cpp +++ b/test/TestDestroy.cpp @@ -27,17 +27,27 @@ TEST(TestDestroy, TestDestroyTensorSingle) { kp::Manager mgr; - tensorA = mgr.tensor({ 0, 0, 0 }); + const std::vector initialValues = {0.0f, 0.0f, 0.0f}; + + tensorA = mgr.tensor(initialValues); std::shared_ptr algo = - mgr.algorithm({ tensorA }, spirv); + mgr.algorithm({tensorA}, spirv); + + // Sync values to and from device + mgr.sequence() + ->eval(algo->getTensors()) + ->eval(algo->getTensors()); + + EXPECT_EQ(tensorA->vector(), initialValues); mgr.sequence() ->record(algo) ->eval() ->eval(algo->getTensors()); - EXPECT_EQ(tensorA->vector(), std::vector({ 1, 1, 1 })); + const std::vector expectedFinalValues = {1.0f, 1.0f, 1.0f}; + EXPECT_EQ(tensorA->vector(), expectedFinalValues); tensorA->destroy(); EXPECT_FALSE(tensorA->isInit());