Moved all functions from hpp to cpp in tensor

src/Tensor.cpp

@@ -70,6 +70,95 @@ Tensor::isInit()
            this->mRawData;
 }
 
+uint32_t
+Tensor::size()
+{
+    return this->mSize;
+}
+
+uint32_t
+Tensor::dataTypeMemorySize()
+{
+    return this->mDataTypeMemorySize;
+}
+
+uint32_t
+Tensor::memorySize()
+{
+    return this->mSize * this->mDataTypeMemorySize;
+}
+
+kp::Tensor::TensorDataTypes
+Tensor::dataType()
+{
+    return this->mDataType;
+}
+
+void*
+Tensor::rawData()
+{
+    return this->mRawData;
+}
+
+void
+Tensor::setRawData(const void* data)
+{
+    memcpy(this->mRawData, data, this->memorySize());
+}
+
+void
+Tensor::mapRawData()
+{
+
+    KP_LOG_DEBUG("Kompute Tensor mapping data from host buffer");
+
+    std::shared_ptr<vk::DeviceMemory> hostVisibleMemory = nullptr;
+
+    if (this->mTensorType == TensorTypes::eHost) {
+        hostVisibleMemory = this->mPrimaryMemory;
+    } else if (this->mTensorType == TensorTypes::eDevice) {
+        hostVisibleMemory = this->mStagingMemory;
+    } else {
+        KP_LOG_WARN(
+          "Kompute Tensor mapping data not supported on storage tensor");
+        return;
+    }
+
+    vk::DeviceSize bufferSize = this->memorySize();
+
+    // Given we request coherent host memory we don't need to invalidate /
+    // flush
+    this->mRawData = this->mDevice->mapMemory(
+      *hostVisibleMemory, 0, bufferSize, vk::MemoryMapFlags());
+
+    vk::MappedMemoryRange mappedMemoryRange(
+      *hostVisibleMemory, 0, bufferSize);
+}
+
+void
+Tensor::unmapRawData()
+{
+
+    KP_LOG_DEBUG("Kompute Tensor mapping data from host buffer");
+
+    std::shared_ptr<vk::DeviceMemory> hostVisibleMemory = nullptr;
+
+    if (this->mTensorType == TensorTypes::eHost) {
+        hostVisibleMemory = this->mPrimaryMemory;
+    } else if (this->mTensorType == TensorTypes::eDevice) {
+        hostVisibleMemory = this->mStagingMemory;
+    } else {
+        KP_LOG_WARN(
+          "Kompute Tensor mapping data not supported on storage tensor");
+        return;
+    }
+
+    vk::DeviceSize bufferSize = this->memorySize();
+    vk::MappedMemoryRange mappedRange(*hostVisibleMemory, 0, bufferSize);
+    this->mDevice->flushMappedMemoryRanges(1, &mappedRange);
+    this->mDevice->unmapMemory(*hostVisibleMemory);
+}
+
 void
 Tensor::recordCopyFrom(const vk::CommandBuffer& commandBuffer,
                        std::shared_ptr<Tensor> copyFromTensor)