error: ‘cudnnSetRNNDescriptor’ was not declared in this scope -- cudnn 8.2.0 ...

[jiapei100]

Environment:

• OS: Ubuntu 20.04
• CUDA: CUDA Driver = CUDART, CUDA Driver Version = 11.2, CUDA Runtime Version = 11.3, NumDevs = 1
• CUDNN: 8.2.0

File cudnn_emitter.cpp, around line 1166,

error: ‘cudnnSetRNNDescriptor’ was not declared in this scope; did you mean ‘cudnnSetLRNDescriptor’?

How to use cudnnSetRNNDescriptor_v8 instead of cudnnSetRNNDescriptor ? For now, my modification is as:

//    CUDNN_SAFE_CALL(cudnnSetRNNDescriptor(*m_ctx->cudnn_handle,
//                                          rnn_desc,
//                                          hidden_size,
//                                          num_layers,
//                                          dropout_desc,
//                                          CUDNN_LINEAR_INPUT, // TO DO: support CUDNN_SKIP_INPUT
//                                          cell_dir,
//                                          mode,
//                                          algo,
//                                          data_type));

    cudnnMathType_t math_type = CUDNN_DEFAULT_MATH;
    cudnnRNNBiasMode_t bias_mode = CUDNN_RNN_DOUBLE_BIAS;
    uint32_t aux_flags = 0;
    int32_t projSize = hidden_size - 2 > 0 ? hidden_size - 2 : hidden_size;
    CUDNN_SAFE_CALL(cudnnSetRNNDescriptor_v8(rnn_desc,
                                             algo,
                                             mode,
                                             bias_mode,
                                             cell_dir,
                                             CUDNN_LINEAR_INPUT,
                                             data_type,
                                             CUDNN_DATA_FLOAT,
                                             math_type,
                                             input_size,
                                             hidden_size,
                                             projSize,
                                             num_layers,
                                             dropout_desc,
                                             aux_flags));

Cheers
Pei