def test_spatial_unspecified():
z={} # problemType definition
conv = Convolution(z, 'ConvolutionForward',
config={'TensorAFormat': 'NCHW',
})
log.debug(conv.printUsage(z))
with pytest.raises(RuntimeError, match="ConvolutionConfig field 'spatial' == None"):
conv.makeProblem(n=64, c=1024, k=256, pcc=conv.cc)
def test_mismatch():
z={} # problemType definition
conv = Convolution(z, 'ConvolutionForward',
config={'TensorAFormat': 'NCHW',
'Spatial' : '34x99',
})
log.debug(conv.printUsage(z))
pcc = ConvolutionConfig(spatial=[14,15])
with pytest.raises(RuntimeError, match="Mismatch between ConvolutionConfig value"):
conv.makeProblem(n=64, c=1024, k=256, pcc=pcc)
def test_spatial_in():
z={} # problemType definition
conv = Convolution(z, 'ConvolutionForward',
config={'TensorAFormat': 'NCHW',
})
log.debug(conv.printUsage(z))
pcc = ConvolutionConfig(spatial=[14,15])
p = conv.makeProblem(n=64, c=1024, k=256, pcc=pcc)
assert(p[0] == [210, 256, 64, 1024])
assert(p[1] == [1, -1, -1])
assert(p[2] == [-1, -1, 0])
def test_stride():
z={} # problemType definition
conv = Convolution(z, 'ConvolutionForward',
config={'TensorAFormat': 'NCHW',
'Spatial' : '13x14',
'Stride' : '2x3',
})
log.debug(conv.printUsage(z))
p = conv.makeProblem(n=64, c=1024, k=256, pcc=conv.cc)
assert(p[0] == [4, 6, 256, 64, 1024])
assert(p[1] == [3, 28, -1, -1])
assert(p[2] == [-1, -1, 0])
def test_spatial_parm():
z = {} # problemType definition
conv = Convolution(z,
'ConvolutionForward',
config={
'TensorAFormat': 'NCHW',
'Spatial': '13x14',
})
log.debug(conv.printUsage(z))
pcc = ConvolutionConfig()
p = conv.makeProblem(n=64, c=1024, k=256, pcc=pcc)
assert (p[0] == [182, 256, 64, 1024])
assert (p[1] == [1, 182, 186368])
assert (p[2] == [-1, -1, 0])
def test_stride_filter_dilated():
z = {} # problemType definition
conv = Convolution(z,
'ConvolutionForward',
config={
'TensorAFormat': 'NCHW',
'Spatial': '13x14',
'Stride': '2x3',
'Filter': '3x4',
'Dilation': '2x3',
})
log.debug(conv.printUsage(z))
p = conv.makeProblem(n=64, c=1024, k=256, pcc=conv.cc)
assert (p[0] == [2, 5, 256, 64, 3, 4, 1024])
assert (p[1] == [3, 28, 3, 28, 182, 186368])
assert (p[2] == [-1, -1, -1, -1, 0])