zixuanweeei commented on issue #15767: FullyConnected op with float64 and 
MKL-DNN fails if gradient are not set in a specific way
URL: 
https://github.com/apache/incubator-mxnet/issues/15767#issuecomment-519126191
 
 
   > Seems that there are no data type check for MKL-DNN stateful RNN 
implementation (see 
https://github.com/apache/incubator-mxnet/blob/master/src/operator/rnn.cc#L226).
 So, when input data is `float64`, mkldnn rnn pass still be executed and then 
`unknown mkldnn type` error will be got.
   
   The execution trace of RNN is maked out as below.
   
   
https://github.com/apache/incubator-mxnet/blob/71861238743fcd8177afe52d1562d9078ac547de/src/operator/rnn.cc#L254
   
   
https://github.com/apache/incubator-mxnet/blob/71861238743fcd8177afe52d1562d9078ac547de/src/operator/nn/mkldnn/mkldnn_base-inl.h#L206-L220

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
[email protected]


With regards,
Apache Git Services

Reply via email to