Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/fairscale.git" did not exist on "8f8f8ef986913ee13c78bf0f066451a5ac62686a"
Commit e9128480 authored by “yuguo”'s avatar “yuguo”
Browse files

fix rnn op bug

parent 992bec46
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
paddle/phi/kernels/gpu/rnn_functor.h
View file @ e9128480
...@@ -99,7 +99,7 @@ class RNNDescriptors { ...@@ -99,7 +99,7 @@ class RNNDescriptors {
// ------------------- cudnn dropout descriptors --------------------- // ------------------- cudnn dropout descriptors ---------------------
size_t state_size; size_t state_size;
bool is_initialized = dropout_state->initialized(); bool is_initialized = dropout_state->initialized();
if (!is_test_ && !is_initialized) { if (!is_initialized) {
#ifdef PADDLE_WITH_HIP #ifdef PADDLE_WITH_HIP
PADDLE_ENFORCE_GPU_SUCCESS( PADDLE_ENFORCE_GPU_SUCCESS(
phi::dynload::miopenDropoutGetStatesSize(handle, &state_size)); phi::dynload::miopenDropoutGetStatesSize(handle, &state_size));
...@@ -114,7 +114,7 @@ class RNNDescriptors { ...@@ -114,7 +114,7 @@ class RNNDescriptors {
dev_ctx.GetPlace(), dev_ctx.GetPlace(),
is_initialized, is_initialized,
dropout_prob_, dropout_prob_,
is_test_ ? nullptr : dropout_state, dropout_state,
seed_, seed_,
state_size); 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