From 409c455d8c02a5d4752cb17d6cb52088174453f2 Mon Sep 17 00:00:00 2001
From: ScheissSchiesser <9285100+ScheissSchiesser@users.noreply.github.com>
Date: Sat, 9 Apr 2022 11:37:51 -0400
Subject: [PATCH] Fix TestDestroy.TestDestroyTensorSingle by first syncing the
 tensor data to the device

Signed-off-by: ScheissSchiesser <9285100+ScheissSchiesser@users.noreply.github.com>
---
 test/TestDestroy.cpp | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

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<float> initialValues = {0.0f, 0.0f, 0.0f};
+
+            tensorA = mgr.tensor(initialValues);
 
             std::shared_ptr<kp::Algorithm> algo =
-              mgr.algorithm({ tensorA }, spirv);
+              mgr.algorithm({tensorA}, spirv);
+
+            // Sync values to and from device
+            mgr.sequence()
+            ->eval<kp::OpTensorSyncDevice>(algo->getTensors())
+            ->eval<kp::OpTensorSyncLocal>(algo->getTensors());
+
+            EXPECT_EQ(tensorA->vector(), initialValues);
 
             mgr.sequence()
               ->record<kp::OpAlgoDispatch>(algo)
               ->eval()
               ->eval<kp::OpTensorSyncLocal>(algo->getTensors());
 
-            EXPECT_EQ(tensorA->vector(), std::vector<float>({ 1, 1, 1 }));
+            const std::vector<float> expectedFinalValues = {1.0f, 1.0f, 1.0f};
+            EXPECT_EQ(tensorA->vector(), expectedFinalValues);
 
             tensorA->destroy();
             EXPECT_FALSE(tensorA->isInit());