|
|
@ -300,9 +300,11 @@ class CudnnLSTMGPUKernel : public framework::OpKernel<T> {
|
|
|
|
}
|
|
|
|
}
|
|
|
|
CudnnRNNCache *cudnn_rnn_cache = nullptr;
|
|
|
|
CudnnRNNCache *cudnn_rnn_cache = nullptr;
|
|
|
|
if (cache_var->IsInitialized()) {
|
|
|
|
if (cache_var->IsInitialized()) {
|
|
|
|
|
|
|
|
// const_cast is usually bad.
|
|
|
|
cudnn_rnn_cache = const_cast<framework::Variable *>(cache_var)
|
|
|
|
cudnn_rnn_cache = const_cast<framework::Variable *>(cache_var)
|
|
|
|
->GetMutable<CudnnRNNCache>();
|
|
|
|
->GetMutable<CudnnRNNCache>();
|
|
|
|
} else {
|
|
|
|
} else {
|
|
|
|
|
|
|
|
// const_cast is usually bad.
|
|
|
|
cudnn_rnn_cache = const_cast<framework::Variable *>(cache_var)
|
|
|
|
cudnn_rnn_cache = const_cast<framework::Variable *>(cache_var)
|
|
|
|
->GetMutable<CudnnRNNCache>();
|
|
|
|
->GetMutable<CudnnRNNCache>();
|
|
|
|
std::random_device rnd;
|
|
|
|
std::random_device rnd;
|
|
|
|