fix gelu compile on CUDA 10 (#39045)

0617a3ed · sneaxiy · GitHub · 0837a2cc · 0617a3ed
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/operators/gelu_op.cu paddle/fluid/operators/gelu_op.cu +1 -1

未找到文件。
--- a/paddle/fluid/operators/gelu_op.cu
+++ b/paddle/fluid/operators/gelu_op.cu
@@ -24,7 +24,7 @@ namespace operators {
 #ifdef __NVCC__
 template <bool FastMode>
 static __device__ __forceinline__ float FP32FastTanh(float x) {
-#if __CUDA_ARCH__ >= 750 && !defined(_WIN32)
+#if __CUDA_ARCH__ >= 750 && CUDA_VERSION >= 11000 && !defined(_WIN32)
  if (FastMode) {
    float y;
    asm("tanh.approx.f32 %0,%1; \n\t" : "=f"(y) : "f"(x));