提交 257bd938 authored 作者: notoraptor's avatar notoraptor

Fix typos, remove useless tests.

上级 e31d548c
...@@ -171,7 +171,7 @@ APPLY_SPECIFIC(conv_fwd)(PyGpuArrayObject *input, PyGpuArrayObject *kerns, ...@@ -171,7 +171,7 @@ APPLY_SPECIFIC(conv_fwd)(PyGpuArrayObject *input, PyGpuArrayObject *kerns,
algo = CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_GEMM; algo = CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_GEMM;
// Algo `small` does not work for a batch size > 2^16, with cuDNN >= V5.1. // Algo `small` does not work for a batch size > 2^16, with cuDNN >= V5.1.
// Issue should have been resolved for cuDNN >= V6.0.20. // Issue should be resolved for cuDNN > V6.0.20.
if (cudnnGetVersion() <= 6020 && if (cudnnGetVersion() <= 6020 &&
algo == CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_PRECOMP_GEMM && algo == CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_PRECOMP_GEMM &&
PyGpuArray_DIM(input, 0) > 65536) PyGpuArray_DIM(input, 0) > 65536)
......
...@@ -1111,17 +1111,17 @@ def run_conv_small_batched_vs_multicall(inputs_shape, filters_shape, batch_sub): ...@@ -1111,17 +1111,17 @@ def run_conv_small_batched_vs_multicall(inputs_shape, filters_shape, batch_sub):
def test_batched_conv_small(): def test_batched_conv_small():
yield (run_conv_small_batched_vs_multicall, (65534, 2, 2, 2), (1, 2, 2, 2), 5) # OK # OK
yield (run_conv_small_batched_vs_multicall, (65535, 2, 2, 2), (1, 2, 2, 2), 5) # OK yield (run_conv_small_batched_vs_multicall, (65536, 2, 2, 2), (1, 2, 2, 2), 5)
yield (run_conv_small_batched_vs_multicall, (65536, 2, 2, 2), (1, 2, 2, 2), 5) # OK # Should fail with cuDNN < V6020, but there's currently a workaround in `dnn_fwd.c` for that case.
yield (run_conv_small_batched_vs_multicall, (65537, 2, 2, 2), (1, 2, 2, 2), 5) # ERROR yield (run_conv_small_batched_vs_multicall, (65537, 2, 2, 2), (1, 2, 2, 2), 5)
def test_batched_conv3d_small(): def test_batched_conv3d_small():
yield (run_conv_small_batched_vs_multicall, (65534, 2, 2, 2, 2), (1, 2, 2, 2, 2), 5) # OK # OK
yield (run_conv_small_batched_vs_multicall, (65535, 2, 2, 2, 2), (1, 2, 2, 2, 2), 5) # OK yield (run_conv_small_batched_vs_multicall, (65536, 2, 2, 2, 2), (1, 2, 2, 2, 2), 5)
yield (run_conv_small_batched_vs_multicall, (65536, 2, 2, 2, 2), (1, 2, 2, 2, 2), 5) # OK # Should fail with cuDNN < V6020, but there's currently a workaround in `dnn_fwd.c` for that case.
yield (run_conv_small_batched_vs_multicall, (65537, 2, 2, 2, 2), (1, 2, 2, 2, 2), 5) # ERROR ALSO. yield (run_conv_small_batched_vs_multicall, (65537, 2, 2, 2, 2), (1, 2, 2, 2, 2), 5)
def test_conv3d_fwd(): def test_conv3d_fwd():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论