diff options
author | Soumith Chintala <soumith@gmail.com> | 2014-09-11 02:04:38 +0400 |
---|---|---|
committer | Soumith Chintala <soumith@gmail.com> | 2014-09-21 03:51:03 +0400 |
commit | 4b3f518fcbc21055f60bb7e18fa5b278d9cc8c79 (patch) | |
tree | 918eb2ac340eb9e21ae250f296f4a575df20c2e5 /test | |
parent | 459a994a8b7f5dfa0a6e984531c053dad234a92e (diff) |
bug repro
Diffstat (limited to 'test')
-rw-r--r-- | test/bug.lua | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/test/bug.lua b/test/bug.lua new file mode 100644 index 0000000..456c261 --- /dev/null +++ b/test/bug.lua @@ -0,0 +1,15 @@ +require 'cutorch' +require 'cudnn' + +local gconv = cudnn.SpatialConvolution(3,16,5,5):cuda() +local inp = torch.randn(128, 3, 100, 100):cuda() +local out = gconv:forward(inp) +-- runs ok +for i=1,100 do + out = gconv:forward(inp) +end +print('ok!') + +print('problemo!') +inp = torch.randn(128, 3, 200, 200):cuda() -- change input shape +out = gconv:forward(inp) |