From 17be1b758a2dc8986c3f76f5092467f191fcfc64 Mon Sep 17 00:00:00 2001 From: Natalia Gimelshein Date: Fri, 22 Apr 2016 17:54:34 -0700 Subject: fix for V5 GA RNN APIs --- test/test_rnn.lua | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'test') 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() -- cgit v1.2.3