提交 32ae3995 authored 作者: Frederic's avatar Frederic

Added gpu conv test shape bigger then maxThreadsDim0

上级 44f44dc3
...@@ -364,6 +364,9 @@ def get_valid_shapes(): ...@@ -364,6 +364,9 @@ def get_valid_shapes():
, ((20, 16, 32, 32), (1, 16, 28, 28), (1,1), (1,1), (1,1)) # layer 1 backprop to weights , ((20, 16, 32, 32), (1, 16, 28, 28), (1,1), (1,1), (1,1)) # layer 1 backprop to weights
, ((60,20,28,28), (10,20,5,5), (1,1), (2,2), (1,1))#added a test case that fail from test_nnet.py.test_conv_nnet2 , ((60,20,28,28), (10,20,5,5), (1,1), (2,2), (1,1))#added a test case that fail from test_nnet.py.test_conv_nnet2
, ((10,5,28,28), (10,5,5,5), (1,1), (2,2), (1,1))#test precedent but reduced that triger the error , ((10,5,28,28), (10,5,5,5), (1,1), (2,2), (1,1))#test precedent but reduced that triger the error
#Test more then maxThreadsDim0
, ((2,4,13,1050), (3,4,10, 11), (1,1), (1,1), (1,1))
, ((2,4,1050,13), (3,4,10, 11), (1,1), (1,1), (1,1))
] ]
shapes += [ ((60,1,28,28),(20,1,5,5), (1,1), (1,1), (1,1))#test_lenet_28 1 layers shapes += [ ((60,1,28,28),(20,1,5,5), (1,1), (1,1), (1,1))#test_lenet_28 1 layers
...@@ -633,6 +636,9 @@ def test_full(): ...@@ -633,6 +636,9 @@ def test_full():
, ((10,30,23,23),(20,30,7,7), (1,1), (1,1), (1,1))#test_lenet_64 full , ((10,30,23,23),(20,30,7,7), (1,1), (1,1), (1,1))#test_lenet_64 full
# , ((20,10,29,29),(30,10,23,23), (1,1), (1,1), (1,1))#test_lenet_64 bprop 1 # , ((20,10,29,29),(30,10,23,23), (1,1), (1,1), (1,1))#test_lenet_64 bprop 1
# , ((1,10,64,64),(20,10,58,58), (1,1), (1,1), (1,1))#test_lenet_64 bprop 2 # , ((1,10,64,64),(20,10,58,58), (1,1), (1,1), (1,1))#test_lenet_64 bprop 2
#Test more then maxThreadsDim0
, ((2,4,13,1050), (3,4,10, 11), (1,1), (1,1), (1,1))
, ((2,4,1050,13), (3,4,10, 11), (1,1), (1,1), (1,1))
] ]
# shapes=shapes[:277] # shapes=shapes[:277]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论