@@ -122,7 +122,10 @@ void findFunc(Options &opts, ModuleOp mod) {
122122 }
123123}
124124
125+ #include < iostream>
126+
125127int main (int argc, char **argv) {
128+ std::cout << " STARTING GC-GPU-RUNNER" << std::endl;
126129 Options opts;
127130 llvm::cl::ParseCommandLineOptions (argc, argv, " GraphCompiler GPU runner\n " );
128131
@@ -136,6 +139,7 @@ int main(int argc, char **argv) {
136139 srcMgr->AddNewSourceBuffer (std::move (file), SMLoc ());
137140 MLIRContext mlirCtx{gc::initCompilerAndGetDialects ()};
138141 auto mlirMod = parseSourceFile<ModuleOp>(srcMgr, {&mlirCtx});
142+ std::cout << " PARSED SOURCE FILE" << std::endl;
139143 findFunc (opts, *mlirMod);
140144
141145 gc::gpu::OclModuleBuilderOpts builderOpts;
@@ -154,8 +158,10 @@ int main(int argc, char **argv) {
154158 };
155159
156160 gc::gpu::OclModuleBuilder builder{mlirMod, builderOpts};
161+ std::cout << " CONSTRUCTED BUILDER" << std::endl;
157162 auto runtime = gcGetOrReport (gc::gpu::OclRuntime::get ());
158163 auto oclMod = gcGetOrReport (builder.build (runtime));
164+ std::cout << " LOWERED MODULE" << std::endl;
159165 assert (oclMod->isStatic );
160166
161167 if (!opts.objDumpFile .getValue ().empty ()) {
@@ -167,7 +173,9 @@ int main(int argc, char **argv) {
167173 gc::gpu::OclContext ctx{runtime, queue};
168174 gc::gpu::StaticExecutor<0 > exec{oclMod};
169175 gcLogD (" Executing function " , opts.mainFuncName .c_str (), " ()" );
176+ std::cout << " STARTING MODULE..." << std::endl;
170177 exec (ctx);
178+ std::cout << " MODULE DONE" << std::endl;
171179 gcGetOrReport (ctx.finish ());
172180 gcGetOrReport (runtime.releaseQueue (queue));
173181 return 0 ;
0 commit comments