@@ -48,7 +48,7 @@ def fun(float(N, M) A, float(N, M) B) -> (C) {
48
48
auto ctx = isl::with_exceptions::globalIslCtx ();
49
49
auto scop = polyhedral::Scop::makeScop (ctx, tc);
50
50
scop = Scop::makeSpecializedScop<int >(*scop, {{" N" , N}, {" M" , M}});
51
- auto mscop = MappedScop::makeSequential (
51
+ auto mscop = cpu:: MappedScop::makeSequential (
52
52
std::move (scop), CpuMappingOptions::makeNaiveMappingOptions ());
53
53
auto pJit = mscop->codegen (" kernel_anon" );
54
54
auto fptr = reinterpret_cast <void (*)(float *, float *, float *, int , int )>(
@@ -89,7 +89,7 @@ TEST(LLVMCodegen, MultiStmt) {
89
89
auto scop = polyhedral::Scop::makeScop (ctx, tc);
90
90
scop = Scop::makeSpecializedScop<int >(
91
91
*scop, {{" N" , N}, {" M" , M}, {" K" , K}, {" L" , L}});
92
- auto mscop = MappedScop::makeSequential (
92
+ auto mscop = cpu:: MappedScop::makeSequential (
93
93
std::move (scop), CpuMappingOptions::makeNaiveMappingOptions ());
94
94
auto pJit = mscop->codegen (" kernel_anon" );
95
95
auto fptr = reinterpret_cast <void (*)(
@@ -170,7 +170,7 @@ def batch_matmul(float(B, N, M) X, float(B, M, K) Y) -> (Z) {
170
170
auto scop = polyhedral::Scop::makeScop (ctx, tc);
171
171
scop = Scop::makeSpecializedScop<int >(
172
172
*scop, {{" N" , N}, {" M" , M}, {" K" , K}, {" B" , B}});
173
- auto mscop = MappedScop::makeSequential (
173
+ auto mscop = cpu:: MappedScop::makeSequential (
174
174
std::move (scop), CpuMappingOptions::makeNaiveMappingOptions ());
175
175
auto pJit = mscop->codegen (" batch_matmul" );
176
176
auto fptr =
@@ -220,7 +220,7 @@ def convolution(float(N,C,H,W) I, float(O,C,KH,KW) W1, float(O) B) -> (tmp, O1)
220
220
{" W" , W},
221
221
{" KW" , KW},
222
222
{" C" , C}});
223
- auto mscop = MappedScop::makeSequential (
223
+ auto mscop = cpu:: MappedScop::makeSequential (
224
224
std::move (scop), CpuMappingOptions::makeNaiveMappingOptions ());
225
225
auto pJit = mscop->codegen (" convolution" );
226
226
auto fptr = reinterpret_cast <void (*)(
@@ -274,7 +274,7 @@ def concat(float(M, N) A, float(M, N) B) -> (O1) {
274
274
auto ctx = isl::with_exceptions::globalIslCtx ();
275
275
auto scop = polyhedral::Scop::makeScop (ctx, tc);
276
276
scop = Scop::makeSpecializedScop<int >(*scop, {{" N" , N}, {" M" , M}});
277
- auto mscop = MappedScop::makeSequential (
277
+ auto mscop = cpu:: MappedScop::makeSequential (
278
278
std::move (scop), CpuMappingOptions::makeNaiveMappingOptions ());
279
279
auto pJit = mscop->codegen (" concat" );
280
280
auto fptr = reinterpret_cast <void (*)(float *, float *, float *, int , int )>(
0 commit comments