remove old data check (#35077)

PaddlePaddle · Aug 23, 2021 · 5b814fd · 5b814fd
1 parent 673bf71
commit 5b814fd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 10 deletions.
diff --git a/paddle/fluid/inference/tensorrt/plugin/emb_eltwise_layernorm_plugin.cu b/paddle/fluid/inference/tensorrt/plugin/emb_eltwise_layernorm_plugin.cu
@@ -139,15 +139,9 @@ int EmbEltwiseLayernormPluginDynamicImpl<T>::enqueue(
   auto emb_ptr_gpu_d =
       emb_ptr_tensor_.mutable_data<int64_t>(platform::CUDAPlace(device_id_));
 
-  auto new_input_ptr = reinterpret_cast<uintptr_t>(inputs[0]);
-
-  if (old_input_ptr_ != new_input_ptr) {
-    old_input_ptr_ = new_input_ptr;
-
-    cudaMemcpyAsync(in_ptr_gpu_d, reinterpret_cast<const void *>(inputs),
-                    sizeof(uintptr_t) * input_num, cudaMemcpyHostToDevice,
-                    stream);
-  }
+  cudaMemcpyAsync(in_ptr_gpu_d, reinterpret_cast<const void *>(inputs),
+                  sizeof(uintptr_t) * input_num, cudaMemcpyHostToDevice,
+                  stream);
 
   auto out_type = output_desc[0].type;
 

diff --git a/paddle/fluid/inference/tensorrt/plugin/emb_eltwise_layernorm_plugin.h b/paddle/fluid/inference/tensorrt/plugin/emb_eltwise_layernorm_plugin.h
@@ -89,7 +89,6 @@ class EmbEltwiseLayernormPluginDynamicImpl
 
   framework::Tensor in_ptr_tensor_, emb_ptr_tensor_;
   int device_id_{0};
-  uintptr_t old_input_ptr_{0};
   bool is_initialized_{false};
 };