@@ -32,15 +32,7 @@ using namespace tc::dlutils;
32
32
33
33
struct TC2Isl : public ::testing::Test {
34
34
void SetUp () {}
35
- void Check (const string& tc, const std::vector<long >& inputSizes) {
36
- auto ctx = getCPUDLContext ();
37
- DLDataType dtype;
38
- dtype.code = kDLFloat ;
39
- dtype.bits = 32 ;
40
- dtype.lanes = 1 ;
41
- auto UPtr = makeDLTensorWithSizes (ctx, dtype, inputSizes);
42
- std::vector<const DLTensor*> inputs{UPtr.get ()};
43
-
35
+ void Check (const string& tc) {
44
36
auto halide =
45
37
tc2halide::translate (isl::with_exceptions::globalIslCtx (), tc);
46
38
auto scop = polyhedral::Scop::makeScop (
@@ -57,7 +49,7 @@ def fun(float(M) I) -> (O) {
57
49
O(i) = I(i)
58
50
}
59
51
)TC" ;
60
- Check (tc, { 123 } );
52
+ Check (tc);
61
53
}
62
54
63
55
TEST_F (TC2Isl, Copy2D) {
@@ -66,7 +58,7 @@ def fun(float(M, N) I) -> (O) {
66
58
O(i, j) = I(i, j)
67
59
}
68
60
)TC" ;
69
- Check (tc, { 123 , 1 } );
61
+ Check (tc);
70
62
}
71
63
72
64
TEST_F (TC2Isl, Copy3D) {
@@ -75,7 +67,7 @@ def fun(float(M, N, P) I) -> (O) {
75
67
O(i, j, k) = I(i, j, k)
76
68
}
77
69
)TC" ;
78
- Check (tc, { 123 , 3 , 2 } );
70
+ Check (tc);
79
71
}
80
72
81
73
TEST_F (TC2Isl, Copy4D) {
@@ -84,7 +76,7 @@ def fun(float(M, N, P, Q) I) -> (O) {
84
76
O(i, j, k, l) = I(i, j, k, l)
85
77
}
86
78
)TC" ;
87
- Check (tc, { 123 , 3 , 4 , 5 } );
79
+ Check (tc);
88
80
}
89
81
90
82
TEST_F (TC2Isl, Copy5D) {
@@ -93,7 +85,7 @@ def fun(float(M, N, P, Q, R) I) -> (O) {
93
85
O(i, j, k, l, m) = I(i, j, k, l, m)
94
86
}
95
87
)TC" ;
96
- Check (tc, { 123 , 10 , 2 , 3 , 4 } );
88
+ Check (tc);
97
89
}
98
90
99
91
// Invalid TC atm
@@ -103,7 +95,7 @@ def fun(float(M) I) -> (O) {
103
95
O(0) +=! I(i)
104
96
}
105
97
)TC" ;
106
- Check (tc, { 123 } );
98
+ Check (tc);
107
99
}
108
100
109
101
TEST_F (TC2Isl, Reduction2D) {
@@ -112,7 +104,7 @@ def fun(float(M, N) I) -> (O) {
112
104
O(i) +=! I(i, j)
113
105
}
114
106
)TC" ;
115
- Check (tc, { 123 , 12 } );
107
+ Check (tc);
116
108
}
117
109
118
110
TEST_F (TC2Isl, Reduction3D) {
@@ -121,7 +113,7 @@ def fun(float(M, N, P) I) -> (O) {
121
113
O(i) +=! I(i, j, k)
122
114
}
123
115
)TC" ;
124
- Check (tc, { 123 , 12 , 16 } );
116
+ Check (tc);
125
117
}
126
118
127
119
TEST_F (TC2Isl, Copy1D2Stmt) {
@@ -131,7 +123,7 @@ def fun(float(M) I) -> (O1, O2) {
131
123
O2(i) = O1(i)
132
124
}
133
125
)TC" ;
134
- Check (tc, { 123 } );
126
+ Check (tc);
135
127
}
136
128
137
129
TEST_F (TC2Isl, Copy2D2Stmt) {
@@ -141,7 +133,7 @@ def fun(float(M, N) I) -> (O1, O2) {
141
133
O2(i, j) = O1(i, j)
142
134
}
143
135
)TC" ;
144
- Check (tc, { 123 , 13 } );
136
+ Check (tc);
145
137
}
146
138
147
139
TEST_F (TC2Isl, Copy2D3Stmt) {
@@ -152,7 +144,7 @@ def fun(float(M, N) I) -> (O1, O2, O3) {
152
144
O3(i, j) = O2(i, j)
153
145
}
154
146
)TC" ;
155
- Check (tc, { 123 , 13 } );
147
+ Check (tc);
156
148
}
157
149
158
150
// Invalid TC atm
@@ -163,7 +155,7 @@ def fun(float(M) I) -> (O1, O2) {
163
155
O2(i) = O1(i)
164
156
}
165
157
)TC" ;
166
- Check (tc, { 123 } );
158
+ Check (tc);
167
159
}
168
160
169
161
TEST_F (TC2Isl, Reduction2D2StmtA) {
@@ -173,7 +165,7 @@ def fun(float(M, N) I) -> (O1, O2) {
173
165
O2(i) = O1(i)
174
166
}
175
167
)TC" ;
176
- Check (tc, { 123 , 13 } );
168
+ Check (tc);
177
169
}
178
170
179
171
TEST_F (TC2Isl, Reduction2D2StmtB) {
@@ -183,7 +175,7 @@ def fun(float(M, N) I) -> (O1, O2) {
183
175
O2(i) +=! O1(i, j)
184
176
}
185
177
)TC" ;
186
- Check (tc, { 123 , 13 } );
178
+ Check (tc);
187
179
}
188
180
189
181
TEST_F (TC2Isl, Reduction2D3Stmt) {
@@ -194,7 +186,7 @@ def fun(float(M, N) I) -> (O1, O2, O3) {
194
186
O3(i) = O2(i)
195
187
}
196
188
)TC" ;
197
- Check (tc, { 123 , 13 } );
189
+ Check (tc);
198
190
}
199
191
200
192
TEST_F (TC2Isl, MutableInput) {
@@ -204,7 +196,7 @@ def foo(float(N) A) -> (B) {
204
196
B(k) +=! A(i) where k in 0:1
205
197
}
206
198
)TC" ;
207
- EXPECT_THROW (Check (tc, { 123 } ), ::lang::ErrorReport);
199
+ EXPECT_THROW (Check (tc), ::lang::ErrorReport);
208
200
}
209
201
int main (int argc, char ** argv) {
210
202
::testing::InitGoogleTest (&argc, argv);
0 commit comments