Skip to content

P
pytensor
提交 ac21919b authored 作者: notoraptor's avatar notoraptor
浏览文件
操作

Update special tests.

上级 6c0f23db
隐藏空白字符变更
内嵌 并排
正在显示 包含 4 行增加 和 4 行删除
theano/gpuarray/tests/check_dnn_conv.py
浏览文件 @ ac21919b
...@@ -761,9 +761,9 @@ class TestDnnConv2D(BaseTestDnnConv): ...@@ -761,9 +761,9 @@ class TestDnnConv2D(BaseTestDnnConv):
border_mode=(1, 0), should_fail=True), border_mode=(1, 0), should_fail=True),
ConvCase.fwd(algo='small', dtype='float32', precision='float32', ConvCase.fwd(algo='small', dtype='float32', precision='float32',
inputs_shape=(65536, 2, 2, 2), filters_shape=(1, 2, 2, 2)), inputs_shape=(65536, 2, 2, 2), filters_shape=(1, 2, 2, 2)),
# NB: Due to current workaround (see dnn_fwd.c), this test won't fail for cuDNN < v6100.
ConvCase.fwd(algo='small', dtype='float32', precision='float32', ConvCase.fwd(algo='small', dtype='float32', precision='float32',
inputs_shape=(65537, 2, 2, 2), filters_shape=(1, 2, 2, 2), inputs_shape=(65537, 2, 2, 2), filters_shape=(1, 2, 2, 2))]
should_fail=(version(raises=False) <= 6020))]
class TestDnnConv3D(BaseTestDnnConv): class TestDnnConv3D(BaseTestDnnConv):
...@@ -779,9 +779,9 @@ class TestDnnConv3D(BaseTestDnnConv): ...@@ -779,9 +779,9 @@ class TestDnnConv3D(BaseTestDnnConv):
special_cases = [ConvCase.fwd(algo='small', dtype='float32', precision='float32', special_cases = [ConvCase.fwd(algo='small', dtype='float32', precision='float32',
inputs_shape=(65536, 2, 2, 2, 2), filters_shape=(1, 2, 2, 2, 2)), inputs_shape=(65536, 2, 2, 2, 2), filters_shape=(1, 2, 2, 2, 2)),
# NB: Due to current workaround (see dnn_fwd.c), this test won't fail for cuDNN < v6100.
ConvCase.fwd(algo='small', dtype='float32', precision='float32', ConvCase.fwd(algo='small', dtype='float32', precision='float32',
inputs_shape=(65537, 2, 2, 2, 2), filters_shape=(1, 2, 2, 2, 2), inputs_shape=(65537, 2, 2, 2, 2), filters_shape=(1, 2, 2, 2, 2))]
should_fail=(version(raises=False) <= 6020))]
def test_true_half_config_support(): def test_true_half_config_support():
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论