diff --git a/src/Tensor.cpp b/src/Tensor.cpp index 7e7af4ca4..6ea92ccd6 100644 --- a/src/Tensor.cpp +++ b/src/Tensor.cpp @@ -258,7 +258,8 @@ Tensor::getStagingMemoryPropertyFlags() { switch (this->mTensorType) { case TensorTypes::eDevice: - return vk::MemoryPropertyFlagBits::eHostVisible; + return vk::MemoryPropertyFlagBits::eHostVisible | + vk::MemoryPropertyFlagBits::eHostCoherent; break; default: throw std::runtime_error("Kompute Tensor invalid tensor type");