extract out pet_not
[pet.git] / tests / matmul.scop
blob56ae5f23697b041b3ffa76e85be42a8cbd5e02f3
1 start: 95
2 end: 277
3 context: '[N, K, M] -> {  : K >= 0 and N >= 0 and N <= 2147483647 and K <= 2147483647
4   and M <= 2147483647 and M >= -2147483648 }'
5 arrays:
6 - context: '[K] -> {  : K >= 0 }'
7   extent: '[N, K, M] -> { A[i0, i1] : i1 >= 0 and i0 >= 0 and i1 <= -1 + K }'
8   element_type: float
9   element_size: 4
10 - context: '[N] -> {  : N >= 0 }'
11   extent: '[N, K, M] -> { B[i0, i1] : i1 >= 0 and i0 >= 0 and i1 <= -1 + N }'
12   element_type: float
13   element_size: 4
14 - context: '[N] -> {  : N >= 0 }'
15   extent: '[N, K, M] -> { C[i0, i1] : i1 >= 0 and i0 >= 0 and i1 <= -1 + N }'
16   element_type: float
17   element_size: 4
18   live_out: 1
19 statements:
20 - line: 9
21   domain: '[N, K, M] -> { S_0[i, j] : j <= -1 + N and j >= 0 and i <= -1 + M and i
22     >= 0 }'
23   schedule: '[N, M] -> { S_0[i, j] -> [0, i, j, 0] }'
24   body:
25     type: op
26     operation: =
27     arguments:
28     - type: access
29       relation: '[N, K, M] -> { S_0[i, j] -> C[i, j] }'
30       index: '[N, K, M] -> { S_0[i, j] -> C[(i), (j)] }'
31       reference: __pet_ref_0
32       read: 0
33       write: 1
34     - type: int
35       value: 0
36 - line: 11
37   domain: '[N, K, M] -> { S_1[i, j, k] : k <= -1 + K and k >= 0 and j <= -1 + N and
38     j >= 0 and i <= -1 + M and i >= 0 }'
39   schedule: '[K, N, M] -> { S_1[i, j, k] -> [0, i, j, 1, k] }'
40   body:
41     type: op
42     operation: +=
43     arguments:
44     - type: access
45       relation: '[N, K, M] -> { S_1[i, j, k] -> C[i, j] }'
46       index: '[N, K, M] -> { S_1[i, j, k] -> C[(i), (j)] }'
47       reference: __pet_ref_1
48       read: 1
49       write: 1
50     - type: op
51       operation: '*'
52       arguments:
53       - type: access
54         relation: '[N, K, M] -> { S_1[i, j, k] -> A[i, k] }'
55         index: '[N, K, M] -> { S_1[i, j, k] -> A[(i), (k)] }'
56         reference: __pet_ref_2
57         read: 1
58         write: 0
59       - type: access
60         relation: '[N, K, M] -> { S_1[i, j, k] -> B[k, j] }'
61         index: '[N, K, M] -> { S_1[i, j, k] -> B[(k), (j)] }'
62         reference: __pet_ref_3
63         read: 1
64         write: 0