add args in create_rnn_descriptor
@@ -296,8 +296,8 @@ def set_dropout_descriptor(desc, handle, dropout):
cudnn.setDropoutDescriptor(desc.value, handle, dropout, 0, 0, 0)
-def create_rnn_descriptor(hidden_size, num_layers, dropout_desc,
- input_mode, direction, mode, data_type):
+def create_rnn_descriptor(handle, hidden_size, num_layers, dropout_desc,
+ input_mode, direction, mode, algo, data_type):
desc = Descriptor(cudnn.createRNNDescriptor(),
cudnn.destroyRNNDescriptor)
if _cudnn_version >= 6000:
