bug fix (#45853)

b971ba04 · wenbin · GitHub · 83cf6758 · b971ba04
显示空白变更内容
内联并排

Showing with 0 addition and 2 deletion

paddle/fluid/inference/tensorrt/plugin/layernorm_shift_partition_op.cu ...inference/tensorrt/plugin/layernorm_shift_partition_op.cu +0 -2

未找到文件。
--- a/paddle/fluid/inference/tensorrt/plugin/layernorm_shift_partition_op.cu
+++ b/paddle/fluid/inference/tensorrt/plugin/layernorm_shift_partition_op.cu
@@ -386,7 +386,6 @@ void invokeLayernormShiftPartition(T *out,
  }
 }
-#if CUDA_ARCH_FP16_SUPPORTED(__CUDA_ARCH__)
 template <>
 void invokeLayernormShiftPartition(half *out,
                                   const half *input,
@@ -433,7 +432,6 @@ void invokeLayernormShiftPartition(half *out,
        eps);
  }
 }
-#endif
 template <typename T>
 static void convertAndCopy(const std::vector<float> &host, T *dev) {