Updated tests to reflect staging

This commit is contained in:
Alejandro Saucedo 2021-02-08 07:18:05 +00:00
parent 04853df469
commit 8a66c6b1e2
4 changed files with 13 additions and 13 deletions

View file

@ -17,9 +17,9 @@ TEST(TestTensor, CopyFromHostData)
std::vector<float> vecB{ 0, 0, 0 };
std::shared_ptr<kp::Tensor> tensorA =
std::make_shared<kp::Tensor>(vecA, kp::Tensor::TensorTypes::eStaging);
std::make_shared<kp::Tensor>(vecA, kp::Tensor::TensorTypes::eHost);
std::shared_ptr<kp::Tensor> tensorB =
std::make_shared<kp::Tensor>(vecB, kp::Tensor::TensorTypes::eStaging);
std::make_shared<kp::Tensor>(vecB, kp::Tensor::TensorTypes::eHost);
kp::Manager mgr;