diff options
| author | Natalia Gimelshein <ngimelshein@nvidia.com> | 2016-04-23 03:54:34 +0300 |
|---|---|---|
| committer | Boris Fomitchev <bfomitchev@nvidia.com> | 2016-05-17 01:12:28 +0300 |
| commit | 17be1b758a2dc8986c3f76f5092467f191fcfc64 (patch) | |
| tree | 8e45d3b61a50cc028897c7cbaeb012b2a13c1777 /test | |
| parent | 8869d166a00769669d5bb1c981c7c02f03388804 (diff) | |
fix for V5 GA RNN APIs
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_rnn.lua | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/test/test_rnn.lua b/test/test_rnn.lua index e7ee3de..2476ce4 100644 --- a/test/test_rnn.lua +++ b/test/test_rnn.lua @@ -216,7 +216,7 @@ function getRNNCheckSums(miniBatch, seqLength, hiddenSize, numberOfLayers, numbe cudnn.getHandle(), rnn.rnnDesc[0], layer, - rnn.xDescs, + rnn.xDescs[0], rnn.wDesc[0], rnn.weight:data(), layerId, @@ -247,7 +247,7 @@ function getRNNCheckSums(miniBatch, seqLength, hiddenSize, numberOfLayers, numbe cudnn.getHandle(), rnn.rnnDesc[0], layer, - rnn.xDescs, + rnn.xDescs[0], rnn.wDesc[0], rnn.weight:data(), layerId, @@ -313,4 +313,4 @@ end mytester = torch.Tester() mytester:add(cudnntest) -mytester:run()
\ No newline at end of file +mytester:run() |