Commit e9128480 authored by “yuguo”'s avatar “yuguo”
Browse files

fix rnn op bug

parent 992bec46
......@@ -99,7 +99,7 @@ class RNNDescriptors {
// ------------------- cudnn dropout descriptors ---------------------
size_t state_size;
bool is_initialized = dropout_state->initialized();
if (!is_test_ && !is_initialized) {
if (!is_initialized) {
#ifdef PADDLE_WITH_HIP
PADDLE_ENFORCE_GPU_SUCCESS(
phi::dynload::miopenDropoutGetStatesSize(handle, &state_size));
......@@ -114,7 +114,7 @@ class RNNDescriptors {
dev_ctx.GetPlace(),
is_initialized,
dropout_prob_,
is_test_ ? nullptr : dropout_state,
dropout_state,
seed_,
state_size);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment