From fdbe114b12fe5306a900bfbddba5bcc50cc5d063 Mon Sep 17 00:00:00 2001
From: zlsh80826 <zlsh80826@gmail.com>
Date: Thu, 21 May 2020 10:05:24 +0800
Subject: [PATCH] [Paddle-TRT] use float constant instead of double
 test=develop (#24544)

---
 paddle/fluid/inference/tensorrt/plugin/gelu_op_plugin.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddle/fluid/inference/tensorrt/plugin/gelu_op_plugin.cu b/paddle/fluid/inference/tensorrt/plugin/gelu_op_plugin.cu
index 6a718d47b15..f94da83b908 100644
--- a/paddle/fluid/inference/tensorrt/plugin/gelu_op_plugin.cu
+++ b/paddle/fluid/inference/tensorrt/plugin/gelu_op_plugin.cu
@@ -63,7 +63,7 @@ __global__ void gelu_kernel(const T a, int n, const T* input, T* output) {
   const int idx = blockIdx.x * TPB + threadIdx.x;
   if (idx < n) {
     const T in = input[idx];
-    const T cdf = 0.5 * (1.0 + erf(in * 0.5 * a));
+    const T cdf = 0.5f * (1.0f + erff(in * 0.5f * a));
     output[idx] = in * cdf;
   }
 }
-- 
GitLab