From 3c486ebf72e192a667cd66df25e1e284027cc9ef Mon Sep 17 00:00:00 2001 From: Alejandro Saucedo Date: Tue, 9 Feb 2021 22:01:26 +0000 Subject: [PATCH] Updated test to cover sequences --- test/TestOpTensorCopy.cpp | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/test/TestOpTensorCopy.cpp b/test/TestOpTensorCopy.cpp index 7b06410..3f2bc95 100644 --- a/test/TestOpTensorCopy.cpp +++ b/test/TestOpTensorCopy.cpp @@ -8,7 +8,7 @@ TEST(TestOpTensorCopy, CopyDeviceToDeviceTensor) kp::Manager mgr; - std::vector testVecA{ 9, 8, 7 }; + std::vector testVecA{ 1, 2, 3 }; std::vector testVecB{ 0, 0, 0 }; std::shared_ptr tensorA{ new kp::Tensor(testVecA) }; @@ -33,7 +33,7 @@ TEST(TestOpTensorCopy, CopyDeviceToDeviceTensorMulti) kp::Manager mgr; - std::vector testVecA{ 9, 8, 7 }; + std::vector testVecA{ 2, 3, 4 }; std::vector testVecB{ 0, 0, 0 }; std::vector testVecC{ 0, 0, 0 }; @@ -63,7 +63,7 @@ TEST(TestOpTensorCopy, CopyDeviceToHostTensor) kp::Manager mgr; - std::vector testVecA{ 9, 8, 7 }; + std::vector testVecA{ 3, 4, 5 }; std::vector testVecB{ 0, 0, 0 }; std::shared_ptr tensorA{ new kp::Tensor(testVecA) }; @@ -92,7 +92,7 @@ TEST(TestOpTensorCopy, CopyHostToDeviceTensor) kp::Manager mgr; - std::vector testVecA{ 9, 8, 7 }; + std::vector testVecA{ 4, 5, 6 }; std::vector testVecB{ 0, 0, 0 }; std::shared_ptr tensorA{ new kp::Tensor( @@ -101,6 +101,9 @@ TEST(TestOpTensorCopy, CopyHostToDeviceTensor) mgr.rebuild({ tensorA, tensorB }, false); + // Manually copy data into host memory of Tensor + tensorA->mapDataIntoHostMemory(); + // Only calling sync on device type tensor mgr.evalOpDefault({ tensorB }); @@ -121,7 +124,7 @@ TEST(TestOpTensorCopy, CopyHostToHostTensor) kp::Manager mgr; - std::vector testVecA{ 9, 8, 7 }; + std::vector testVecA{ 5, 6, 7 }; std::vector testVecB{ 0, 0, 0 }; std::shared_ptr tensorA{ new kp::Tensor( @@ -148,7 +151,7 @@ TEST(TestOpTensorCopy, SingleTensorShouldFail) kp::Manager mgr; - std::vector testVecA{ 9, 8, 7 }; + std::vector testVecA{ 6, 7, 8 }; std::shared_ptr tensorA{ new kp::Tensor( testVecA, kp::Tensor::TensorTypes::eHost) };