Add test for reserved tensor

Reserved tensor now has a proper test, allocating 3 uint32_t. Signed-off-by: Andrea Calabrese <[email protected]>
KomputeProject · Aug 7, 2024 · 8112ed1 · 8112ed1
1 parent 41a21fa
commit 8112ed1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/src/include/kompute/Tensor.hpp b/src/include/kompute/Tensor.hpp
@@ -97,7 +97,8 @@ class Tensor
                  uint32_t elementMemorySize);
 
     /**
-     * @brief Reserve memory on the tensor
+     * Function to reserve memory on the tensor. This does not copy any data, it
+     * just reserves memory, similarly to std::vector reserve() method.
      *
      * @param newSize the new size for reservation
      */
@@ -336,8 +337,7 @@ class TensorT : public Tensor
                this->dataType(),
                tensorType)
     {
-        KP_LOG_DEBUG("Kompute TensorT constructor with data size {}",
-                     data.size());
+        KP_LOG_DEBUG("Kompute TensorT constructor with data size {}", size);
     }
 
     TensorT(std::shared_ptr<vk::PhysicalDevice> physicalDevice,

diff --git a/test/TestTensor.cpp b/test/TestTensor.cpp
@@ -15,6 +15,14 @@ TEST(TestTensor, ConstructorData)
     EXPECT_EQ(tensor->vector(), vec);
 }
 
+TEST(TestTensor, ReserveData)
+{
+    kp::Manager mgr;
+    std::shared_ptr<kp::TensorT<float>> tensor = mgr.tensor(3, sizeof(uint32_t), Tensor::TensorDataType::eUnsignedInt);
+    EXPECT_EQ(tensor->size(), 0);
+    EXPECT_EQ(tensor->dataTypeMemorySize(), sizeof(uint32_t));
+}
+
 TEST(TestTensor, DataTypes)
 {
     kp::Manager mgr;