def cast(poly_sch, fuzz_shape=None): test_ms_cast((32, 32, 14, 14, 16), "float16", "float32", poly_sch=poly_sch) test_ms_cast((32, 32, 14, 14, 16), "float32", "float16", poly_sch=poly_sch)
def test_cast(): test_ms_cast((32, 32, 14, 14, 16), "float16", "float32", poly_sch=True) test_ms_cast((32, 32, 14, 14, 16), "float32", "float16", poly_sch=True) return True