blas-array2 0.3.0

Parameter-optional BLAS wrapper by ndarray::Array (Ix1 or Ix2).
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

import sys
sys.path.append("../")

from taguchi import taguchi_by_list


def gen_valid_owned():
    # test_macro!(test_000: inline, f32, (7, 8, 1, 1), (8, 9, 1, 1), 'R', 'R', 'N', 'N');
    list_num = ["f32", "f64", "c32", "c64"]
    list_layout = ["R", "C"]
    list_stride = [1, 3]
    list_shape_a = [(7, 8, "N"), (8, 7, "T")]
    list_shape_b = [(8, 9, "N"), (9, 8, "T")]

    set_inp = [
        list_num,
        list_shape_a, list_shape_b,
        list_stride, list_stride, list_stride, list_stride,
        list_layout, list_layout,
    ]
    run_size = 16

    tokens = []
    for n, list_taguchi in enumerate(taguchi_by_list(set_inp, run_size)):
        (
            num,
            a_shape, b_shape,
            a_stride_0, a_stride_1, b_stride_0, b_stride_1,
            a_layout, b_layout,
        ) = list_taguchi
        token = (
            f"test_macro!(test_{n:03d}: inline, {num}, "
            f"{(a_shape[0], a_shape[1], a_stride_0, a_stride_1)}, "
            f"{(b_shape[0], b_shape[1], b_stride_0, b_stride_1)}, "
            f"'{a_layout}', '{b_layout}', "
            f"'{a_shape[2]}', '{b_shape[2]}');"
        )
        tokens.append(token)
    return tokens


def gen_valid_view():
    # test_macro!(test_000: inline, f32, (7, 8, 1, 1), (8, 9, 1, 1), (7, 9, 1, 1), 'R', 'R', 'R', 'N', 'N');
    list_num = ["f32", "f64", "c32", "c64"]
    list_layout = ["R", "C"]
    list_stride = [1, 3]
    list_shape_a = [(7, 8, "N"), (8, 7, "T")]
    list_shape_b = [(8, 9, "N"), (9, 8, "T")]

    set_inp = [
        list_num,
        list_shape_a, list_shape_b,
        list_stride, list_stride, list_stride, list_stride, list_stride, list_stride,
        list_layout, list_layout, list_layout,
    ]
    run_size = 16

    tokens = []
    for n, list_taguchi in enumerate(taguchi_by_list(set_inp, run_size)):
        (
            num,
            a_shape, b_shape,
            a_stride_0, a_stride_1, b_stride_0, b_stride_1, c_stride_0, c_stride_1,
            a_layout, b_layout, c_layout,
        ) = list_taguchi
        token = (
            f"test_macro!(test_{n:03d}: inline, {num}, "
            f"{(a_shape[0], a_shape[1], a_stride_0, a_stride_1)}, "
            f"{(b_shape[0], b_shape[1], b_stride_0, b_stride_1)}, "
            f"{(7, 9, c_stride_0, c_stride_1)}, "
            f"'{a_layout}', '{b_layout}', '{c_layout}', "
            f"'{a_shape[2]}', '{b_shape[2]}');"
        )
        tokens.append(token)
    return tokens


def gen_valid_cblas():
    list_layout = ["R", "C"]
    list_shape_a = [(7, 8, "N"), (8, 7, "T"), (8, 7, "C")]
    list_shape_b = [(8, 9, "N"), (9, 8, "T"), (9, 8, "C")]

    set_inp = [
        list_shape_a, list_shape_b,
        list_layout
    ]
    run_size = 12

    tokens = []
    for n, list_taguchi in enumerate(taguchi_by_list(set_inp, run_size, strength=1)):
        (
            (ad0, ad1, transa), (bd0, bd1, transb),
            layout
        ) = list_taguchi
        token = (
            f"test_macro!(test_{n:03d}: inline, c32, cblas_cgemm, "
            f"{(ad0, ad1, 1, 1)}, "
            f"{(bd0, bd1, 1, 1)}, "
            f"{(7, 9, 1, 1)}, "
            f"'{layout}', '{layout}', '{layout}', "
            f"'{transa}', '{transb}', '{layout}');"
        )
        tokens.append(token)
    return tokens


if __name__ == "__main__":
    # print("\n".join(gen_valid_owned()))
    # print("\n".join(gen_valid_view()))
    print("\n".join(gen_valid_cblas()))