projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 39d50ef) | patch
Fixed MIOpen RNN Segfault issue and enabled RNN test (#14810)
authorlcskrishna <lollachaitanya@gmail.com>
Thu, 6 Dec 2018 07:52:42 +0000 (23:52 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 6 Dec 2018 07:54:31 +0000 (23:54 -0800)
commit12addc64a6a6787c224e4d96057a797ca7de1535
treef8181e044da6606c10d3d5b1aeefd382fd7c2ffbtree | snapshot
parent39d50ef4f69051105a055f8fa7cc4a9cf27eec33commit | diff
Fixed MIOpen RNN Segfault issue and enabled RNN test (#14810)

Summary:
This pull request contains changes for:
1. Added MIOpen RNN API miopenGetRNNLayerBiasSize and miopenGetRNNLayerParamSize.
2. Fixed usage of API miopenGetRNNLayerParam.
3. Modifying the RNN test to run using MIOpen engine.

Differential Revision: D13355699

Pulled By: bddppq

fbshipit-source-id: 6f750657f8049c5446eca893880b397804120b69
caffe2/operators/rnn/hip/recurrent_op_miopen.hip diff | blob | history
caffe2/python/operator_test/cudnn_recurrent_test.py diff | blob | history
Domain: Machine Learning / ML Framework;