Skip to content

Commit 93754dc

Browse files
committed
more logs...
Signed-off-by: dchigarev <dmitry.chigarev@intel.com>
1 parent ac360cd commit 93754dc

File tree

2 files changed

+9
-0
lines changed

2 files changed

+9
-0
lines changed

src/gc-gpu-runner/GpuRunner.cpp

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,10 @@ void findFunc(Options &opts, ModuleOp mod) {
122122
}
123123
}
124124

125+
#include <iostream>
126+
125127
int 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;

test/mlir/unittests/ExecutionEngine/GPU/GpuOclRuntimeTest.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626
#include "llvm/Support/SourceMgr.h"
2727

2828
#include "gtest/gtest.h"
29+
#include <iostream>
2930

3031
using namespace mlir;
3132
using namespace gc::gpu;

0 commit comments

Comments
 (0)