Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
未验证 提交 1dd7b97b 编写于 作者: J Jack Zhou 提交者： GitHub
浏览文件
操作

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

上级 87bb7262
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
paddle/fluid/platform/cudnn_helper.h
浏览文件 @ 1dd7b97b
...@@ -314,7 +314,7 @@ class ScopedRNNTensorDescriptor { ...@@ -314,7 +314,7 @@ class ScopedRNNTensorDescriptor {
inline cudnnRNNDataDescriptor_t descriptor( inline cudnnRNNDataDescriptor_t descriptor(
const cudnnDataType_t cudnn_type, int max_seq_length, int batch_size, const cudnnDataType_t cudnn_type, int max_seq_length, int batch_size,
int input_size, bool time_major, const std::vector<int>& seq_length) { 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; cudnnRNNDataLayout_t layout;
if (time_major) { if (time_major) {
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或