From 2eebed93cb6f496692660894fa0533dc0544e5c8 Mon Sep 17 00:00:00 2001 From: Sheng Zha Date: Wed, 18 Jul 2018 21:08:50 -0700 Subject: [PATCH] enable CPU kernel for all RNN layer forward (#11807) --- python/mxnet/gluon/rnn/rnn_layer.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/python/mxnet/gluon/rnn/rnn_layer.py b/python/mxnet/gluon/rnn/rnn_layer.py index cda91380763c..418c497ce832 100644 --- a/python/mxnet/gluon/rnn/rnn_layer.py +++ b/python/mxnet/gluon/rnn/rnn_layer.py @@ -185,11 +185,7 @@ def forward(self, inputs, states=None): for i in range(self._dir): self.i2h_weight[i].shape = (self._gates*self._hidden_size, inputs.shape[2]) self.i2h_weight[i]._finish_deferred_init() - if inputs.context.device_type == 'gpu' or \ - self._mode in ['lstm', 'gru'] and not self._dropout: - out = self._forward_kernel(inputs, states) - else: - out = self._forward(inputs, states) + out = self._forward_kernel(inputs, states) # out is (output, state) return out[0] if skip_states else out