提交 7a95b2a4 authored 作者: f0k's avatar f0k

Test if subsampling GpuConv ops get properly replaced with CorrMM ops

上级 7294d64c
......@@ -742,10 +742,20 @@ def test_subsample():
if ones:
random = False
# exec_conv(version_valid, shapes, verbose, random, 'valid',
# print_=print_, ones=ones)
# exec_conv(version_full, shapes, verbose, random, 'full',
# print_=print_, ones=ones)
# Test the GpuCorrMM version
mode = theano_mode.including("conv_gemm")
exec_conv(version_valid, shapes, verbose, random, 'valid',
print_=print_, ones=ones)
print_=print_, ones=ones,
theano_mode=mode, cls=cuda.blas.GpuCorrMM)
exec_conv(version_full, shapes, verbose, random, 'full',
print_=print_, ones=ones)
print_=print_, ones=ones,
theano_mode=mode, cls=cuda.blas.GpuCorrMM)
class TestConv2DGPU(unittest.TestCase):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论