From 1dd7b97b66078daf133da69e6c611951c98694b2 Mon Sep 17 00:00:00 2001 From: Jack Zhou Date: Mon, 7 Dec 2020 15:47:59 +0800 Subject: [PATCH] fix rnn_op bug in cudnn_version>= 8 (#29406) --- paddle/fluid/platform/cudnn_helper.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/platform/cudnn_helper.h b/paddle/fluid/platform/cudnn_helper.h index e591852cc95..af0df2efc5e 100644 --- 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& seq_length) { - static float padding_fill = 0.0f; + static double padding_fill = 0.0f; cudnnRNNDataLayout_t layout; if (time_major) { -- GitLab