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 6, 2024 · 10bbd41 · 10bbd41
1 parent 72bbb69
commit 10bbd41
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/include/kompute/Tensor.hpp b/src/include/kompute/Tensor.hpp
@@ -336,8 +336,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;