未验证 提交 1dd7b97b 编写于 作者: J Jack Zhou 提交者: GitHub

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

上级 87bb7262
...@@ -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.
先完成此消息的编辑!
想要评论请 注册