Python test_ms_round示例

示例#1

def round(poly_sch, fuzz_shape=None):
    test_ms_round((1024, 1024), "float32", poly_sch=poly_sch)
    test_ms_round((1024, 1024), "float16", poly_sch=poly_sch)
    test_ms_round((1, ), "float32", poly_sch=poly_sch)
    test_ms_round((1, 1), "float32", poly_sch=poly_sch)
    test_ms_round((1, ), "float16", poly_sch=poly_sch)
    test_ms_round((1, 1), "float16", poly_sch=poly_sch)

示例#2

def test_round():
    test_ms_round((1024, 1024), "float32", poly_sch=True)
    test_ms_round((1024, 1024), "float16", poly_sch=True)
    test_ms_round((1, ), "float32", poly_sch=True)
    test_ms_round((1, 1), "float32", poly_sch=True)
    test_ms_round((1, ), "float16", poly_sch=True)
    test_ms_round((1, 1), "float16", poly_sch=True)
    return True