def add(poly_sch, fuzz_shape=None):
if fuzz_shape:
test_ms_add(fuzz_shape, fuzz_shape, 'float32', poly_sch=poly_sch)
return
test_ms_add((1, 1024), (1, 1024), 'float32', poly_sch=poly_sch)
test_ms_add((2, 32, 256, 32, 32), (2, 32, 256, 32, 32),
'float32', poly_sch=poly_sch)
def test_add():
test_ms_add((1, 1024), (1, 1024), 'float32', poly_sch=True)
test_ms_add((2, 32, 256, 32, 32), (2, 32, 256, 32, 32),
'float32',
poly_sch=True)
return True