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

@ -126,7 +126,7 @@ TEST(TestManager, TestCreateInitTensor)
EXPECT_EQ(tensorB->data(), std::vector<float>({ 0, 1, 2 }));
std::shared_ptr<kp::Tensor> tensorC =
mgr.buildTensor({ 0, 0, 0 }, kp::Tensor::TensorTypes::eStaging);
mgr.buildTensor({ 0, 0, 0 }, kp::Tensor::TensorTypes::eHost);
mgr.evalOpDefault<kp::OpTensorCopy>({ tensorA, tensorC });