kompute/test/TestTensor.cpp

#include "gtest/gtest.h"

#include "kompute/Kompute.hpp"

TEST(TestTensor, ConstructorData)
{
    std::vector<float> vec{ 0, 1, 2 };
    kp::Tensor tensor(vec);
    EXPECT_EQ(tensor.size(), vec.size());
    EXPECT_EQ(tensor.data(), vec);
}

TEST(TestTensor, CopyFromHostData)
{
    std::vector<float> vecA{ 0, 1, 2 };
    std::vector<float> vecB{ 0, 0, 0 };

    std::shared_ptr<kp::Tensor> tensorA =
      std::make_shared<kp::Tensor>(vecA, kp::Tensor::TensorTypes::eStaging);
    std::shared_ptr<kp::Tensor> tensorB =
      std::make_shared<kp::Tensor>(vecB, kp::Tensor::TensorTypes::eStaging);

    kp::Manager mgr;

    if (std::shared_ptr<kp::Sequence> sq =
          mgr.getOrCreateManagedSequence("new").lock()) {
        sq->begin();

        sq->record<kp::OpTensorCreate>({ tensorA, tensorB });

        sq->record<kp::OpTensorCopy>({ tensorA, tensorB });

        sq->end();

        sq->eval();
    }

    EXPECT_EQ(tensorA->data(), tensorB->data());
}