Skip to content

Commit

Permalink
Added logging to brgemm executors
Browse files Browse the repository at this point in the history
  • Loading branch information
v-Golubev committed Jan 30, 2025
1 parent 824f6cc commit e74ab6d
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -328,7 +328,9 @@ void BrgemmBaseKernelExecutor::execute_brgemm_kernel(
brgemm_p.skip_accm = 0;
brgemm_p.BS = 1; // default value
OV_CPU_JIT_EMITTER_ASSERT(kernel, "has nullptr Brgemm kernel");
std::cout << "[ INFO ] BrgemmBaseKernelExecutor is executed...\n";
(*kernel)(&brgemm_p);
std::cout << "[ INFO ] BrgemmBaseKernelExecutor execution is finished.\n";
}

#undef DIM_CAST
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -391,7 +391,10 @@ void BrgemmCopyBKernelExecutor::execute(const BrgemmCopyBKernelExecutor* executo
auto kernel = executor->get_kernel();
OV_CPU_JIT_EMITTER_ASSERT(kernel, "has nullptr kernel");
OV_CPU_JIT_EMITTER_ASSERT(args, "has nullptr call args");
std::cout << "[ INFO ] BrgemmCopyBKernelExecutor is executed...\n";
std::cout << "\t config = " << executor->get_config().to_string() << std::endl;
(*kernel)(args);
std::cout << "[ INFO ] BrgemmCopyBKernelExecutor execution is finished.\n";
}

} // namespace intel_cpu
Expand Down

0 comments on commit e74ab6d

Please sign in to comment.