114 lines · plain
1// RUN: mlir-opt -allow-unregistered-dialect %s | FileCheck %s2 3// CHECK-DAG: #set{{[0-9]*}} = affine_set<() : (0 == 0)>4#set0 = affine_set<() : ()>5 6// CHECK-DAG: #set{{[0-9]*}} = affine_set<(d0) : (d0 == 0)>7#set1 = affine_set<(i) : (i == 0)>8 9// CHECK-DAG: #set{{[0-9]*}} = affine_set<(d0) : (d0 >= 0)>10#set2 = affine_set<(i) : (i >= 0)>11 12// CHECK-DAG: #set{{[0-9]*}} = affine_set<(d0, d1) : (d0 >= 0, d1 >= 0)>13#set3 = affine_set<(i, j) : (i >= 0, j >= 0)>14 15// CHECK-DAG: #set{{[0-9]*}} = affine_set<(d0, d1) : (d0 == 0, d1 >= 0)>16#set4 = affine_set<(i, j) : (i == 0, j >= 0)>17 18// CHECK-DAG: #set{{[0-9]*}} = affine_set<(d0, d1)[s0, s1] : (d0 * 2 + s0 >= 0, d1 * 2 + s1 >= 0, d0 >= 0, d1 >= 0, s0 == 0, s1 == 0)>19#set5 = affine_set<(i, j)[N, M] : (i * 2 + N >= 0, j * 2 + M >= 0, i >= 0, j >= 0, N == 0, M == 0)>20 21// CHECK-DAG: #set{{[0-9]*}} = affine_set<(d0, d1)[s0, s1] : (-d0 + s0 >= 0, -d1 + s1 >= 0, d0 >= 0, d1 >= 0)>22#set6 = affine_set<(i, j)[N, M] : (N - i >= 0, M - j >= 0, i >= 0, j >= 0)>23 24// Check if affine constraints with affine exprs on RHS can be parsed.25 26// CHECK-DAG: #set{{[0-9]*}} = affine_set<(d0) : (d0 - 1 == 0)>27#set7 = affine_set<(i) : (i == 1)>28 29// CHECK-DAG: #set{{[0-9]*}} = affine_set<(d0)[s0, s1] : (d0 >= 0, -d0 + s0 >= 0, s0 - 5 == 0, -d0 + s1 + 1 >= 0)>30#set8 = affine_set<(i)[N, M] : (i >= 0, N >= i, N == 5, M + 1 >= i)>31 32// CHECK-DAG: #set{{[0-9]*}} = affine_set<(d0, d1)[s0] : (d0 >= 0, -d0 + s0 >= 0, d1 >= 0, d0 - d1 >= 0)>33#set9 = affine_set<(i, j)[N] : (i >= 0, N >= i, j >= 0, i >= j)>34 35// CHECK-DAG: #set{{[0-9]*}} = affine_set<(d0, d1)[s0, s1] : (-(d0 + d1 + s0 + s1) == 0, d0 + d1 - (s0 + s1) == 0)>36#set10 = affine_set<(i0, i1)[N, M] : (0 == i0 + i1 + N + M, i0 + i1 == N + M)>37 38// CHECK-DAG: #set{{[0-9]*}} = affine_set<(d0, d1)[s0, s1] : (-(d0 + d1 + s0 + s1) >= 0, d0 + d1 - (s0 + s1) >= 0)>39#set11 = affine_set<(i0, i1)[N, M] : (0 >= i0 + i1 + N + M, i0 + i1 >= N + M)>40 41// CHECK-DAG: #set{{[0-9]*}} = affine_set<(d0, d1, d2, d3) : ((d0 + d1) mod 2 - (d2 + d3) floordiv 2 == 0, d0 mod 2 + d1 mod 2 - (d2 + d3 + d2) >= 0)>42#set12 = affine_set<(d0, d1, r0, r1) : ((d0 + d1) mod 2 == (r0 + r1) floordiv 2, ((d0) mod 2) + ((d1) mod 2) >= (r0 + r1) + r0)>43 44// Check if affine constraints with <= can be parsed.45 46// CHECK-DAG: #set{{[0-9]*}} = affine_set<(d0) : (-d0 >= 0)>47#set13 = affine_set<(i) : (i <= 0)>48 49// CHECK-DAG: #set{{[0-9]*}} = affine_set<(d0) : (d0 >= 0, -d0 + 10 >= 0)>50#set14 = affine_set<(i) : (i >= 0, i <= 10)>51 52// CHECK-DAG: #set{{[0-9]*}} = affine_set<(d0, d1)[s0, s1] : (d0 >= 0, -d0 + s0 >= 0, d1 >= 0, -d1 + s1 >= 0)>53#set15 = affine_set<(i, j)[N, M] : (i >= 0, i <= N, j >= 0, j <= M)>54 55// CHECK-DAG: #set{{[0-9]*}} = affine_set<(d0)[s0] : (d0 >= 0, -d0 + s0 >= 0, -d0 + s0 floordiv 2 >= 0)>56#set16 = affine_set<(i)[N] : (i >= 0, i <= N, i <= N floordiv 2)>57 58// CHECK-DAG: #set{{[0-9]*}} = affine_set<(d0, d1, d2, d3, d4) : (-d0 >= 0, -d1 >= 0, -d2 >= 0, -d3 >= 0, -d4 >= 0)>59#set17 = affine_set<(i0, i1, i2, i3, i4)[] : (i0 <= 0, i1 <= 0, i2 <= 0, i3 <= 0, i4 <= 0)>60 61// CHECK-DAG: "testset0"() {set = #set{{[0-9]*}}} : () -> ()62"testset0"() {set = #set0} : () -> ()63 64// CHECK-DAG: "testset1"() {set = #set{{[0-9]*}}} : () -> ()65"testset1"() {set = #set1} : () -> ()66 67// CHECK-DAG: "testset2"() {set = #set{{[0-9]*}}} : () -> ()68"testset2"() {set = #set2} : () -> ()69 70// CHECK-DAG: "testset3"() {set = #set{{[0-9]*}}} : () -> ()71"testset3"() {set = #set3} : () -> ()72 73// CHECK-DAG: "testset4"() {set = #set{{[0-9]*}}} : () -> ()74"testset4"() {set = #set4} : () -> ()75 76// CHECK-DAG: "testset5"() {set = #set{{[0-9]*}}} : () -> ()77"testset5"() {set = #set5} : () -> ()78 79// CHECK-DAG: "testset6"() {set = #set{{[0-9]*}}} : () -> ()80"testset6"() {set = #set6} : () -> ()81 82// CHECK-DAG: "testset7"() {set = #set{{[0-9]*}}} : () -> ()83"testset7"() {set = #set7} : () -> ()84 85// CHECK-DAG: "testset8"() {set = #set{{[0-9]*}}} : () -> ()86"testset8"() {set = #set8} : () -> ()87 88// CHECK-DAG: "testset9"() {set = #set{{[0-9]*}}} : () -> ()89"testset9"() {set = #set9} : () -> ()90 91// CHECK-DAG: "testset10"() {set = #set{{[0-9]*}}} : () -> ()92"testset10"() {set = #set10} : () -> ()93 94// CHECK-DAG: "testset11"() {set = #set{{[0-9]*}}} : () -> ()95"testset11"() {set = #set11} : () -> ()96 97// CHECK-DAG: "testset12"() {set = #set{{[0-9]*}}} : () -> ()98"testset12"() {set = #set12} : () -> ()99 100// CHECK-DAG: "testset13"() {set = #set{{[0-9]*}}} : () -> ()101"testset13"() {set = #set13} : () -> ()102 103// CHECK-DAG: "testset14"() {set = #set{{[0-9]*}}} : () -> ()104"testset14"() {set = #set14} : () -> ()105 106// CHECK-DAG: "testset15"() {set = #set{{[0-9]*}}} : () -> ()107"testset15"() {set = #set15} : () -> ()108 109// CHECK-DAG: "testset16"() {set = #set{{[0-9]*}}} : () -> ()110"testset16"() {set = #set16} : () -> ()111 112// CHECK-DAG: "testset17"() {set = #set{{[0-9]*}}} : () -> ()113"testset17"() {set = #set17} : () -> ()114