From 7a89a4a5738e4d5b76d0d95cc2dfbcc45bd637ee Mon Sep 17 00:00:00 2001
From: perseusdg <harshvardhan.chandira@gmail.com>
Date: Thu, 28 Oct 2021 23:43:12 +0530
Subject: [PATCH] ReshapeRT.cpp fix

---
 src/pluginsRT/ReshapeRT.cpp | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/pluginsRT/ReshapeRT.cpp b/src/pluginsRT/ReshapeRT.cpp
index dbe59c0..43cf8af 100644
--- a/src/pluginsRT/ReshapeRT.cpp
+++ b/src/pluginsRT/ReshapeRT.cpp
@@ -53,11 +53,10 @@ int ReshapeRT::enqueue(int batchSize, const void *const *inputs, void *const *ou
 }
 #elif NV_TENSORRT_MAJOR <= 7
 int32_t ReshapeRT::enqueue(int32_t batchSize, const void *const *inputs, void **outputs, void *workspace, cudaStream_t stream) {
-    std::cout << new_dim.c << ":" << new_dim.h << std::endl;
     dnnType *srcData = (dnnType*)reinterpret_cast<const dnnType*>(inputs[0]);
     dnnType *dstData = reinterpret_cast<dnnType*>(outputs[0]);
     std::cout << "C : " << c << "H : " << h << "w :" << w << std::endl;
-    checkCuda( cudaMemcpyAsync(dstData, srcData, batchSize*new_dim.c*new_dim.h*new_dim.w*sizeof(dnnType), cudaMemcpyDeviceToDevice, stream));
+    checkCuda( cudaMemcpyAsync(dstData, srcData, batchSize*c*h*w*sizeof(dnnType), cudaMemcpyDeviceToDevice, stream));
     return 0;
 }
 #endif