Updated tests to reflect staging
This commit is contained in:
parent
04853df469
commit
8a66c6b1e2
4 changed files with 13 additions and 13 deletions
|
|
@ -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 });
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue