From 0a4677bdb0399a3012f12a92bb3792e516ecbd0b Mon Sep 17 00:00:00 2001
From: Elizabeth <hashim.sharif91@gmail.com>
Date: Thu, 31 Oct 2019 23:34:58 -0500
Subject: [PATCH] Fixing tensor sizes assertion
---
.../tensor_runtime/include/tensor_utils.cu | 12 +++++++++---
1 file changed, 9 insertions(+), 3 deletions(-)
diff --git a/llvm/projects/hpvm-tensor-rt/tensor_runtime/include/tensor_utils.cu b/llvm/projects/hpvm-tensor-rt/tensor_runtime/include/tensor_utils.cu
index 13ef262bac..282a0cbb68 100644
--- a/llvm/projects/hpvm-tensor-rt/tensor_runtime/include/tensor_utils.cu
+++ b/llvm/projects/hpvm-tensor-rt/tensor_runtime/include/tensor_utils.cu
@@ -329,12 +329,18 @@ extern "C"{
void initTensorData(void* tensor_ptr, void* data_ptr, size_t size_in_bytes){
Tensor* tensor = (Tensor*) tensor_ptr;
-
- if(tensor->size_in_bytes != size_in_bytes){
+
+ size_t host_size_in_bytes = tensor->num_elems * 4;
+ //if(tensor->size_in_bytes != size_in_bytes){
+ if(host_size_in_bytes != size_in_bytes){
ERROR("The destination and source sizes don't match");
}
std::memcpy(tensor->host_data, data_ptr, size_in_bytes);
+
+ changeTensorPlacement(tensor, HOST);
+
+ tensor->cur_type = float_type;
}
@@ -428,7 +434,7 @@ extern "C"{
-bool ONLINE_PROFILING = false;
+bool ONLINE_PROFILING = false; // true;
void convertToFP16(struct Tensor* tensor){
--
GitLab