fix rnn_op bug in cudnn_version>= 8 (#29406)

4 years ago · 1dd7b97b66
parent 87bb726258
commit 1dd7b97b66
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/platform/cudnn_helper.h
+++ b/paddle/fluid/platform/cudnn_helper.h
@ -314,7 +314,7 @@ class ScopedRNNTensorDescriptor {
  inline cudnnRNNDataDescriptor_t descriptor(
      const cudnnDataType_t cudnn_type, int max_seq_length, int batch_size,
      int input_size, bool time_major, const std::vector<int>& seq_length) {
-    static float padding_fill = 0.0f;
+    static double padding_fill = 0.0f;
    cudnnRNNDataLayout_t layout;

    if (time_major) {