Skip to content

Commit fb62cad

Browse files
usstqEgorDuplensky
authored andcommitted
Add log to jit dump code
1 parent bc48caa commit fb62cad

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/cpu/jit_utils/jit_utils.cpp

+5
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
*******************************************************************************/
1717

1818
#include <mutex>
19+
#include <iostream>
1920

2021
#include "common/utils.hpp"
2122

@@ -57,6 +58,7 @@ void dump_jit_code(const void *code, size_t code_size, const char *code_name) {
5758
counter);
5859
counter++;
5960

61+
std::cout << "[ oneDNN ] dump_jit_code: " << fname << std::endl;
6062
FILE *fp = fopen(fname, "wb+");
6163
// Failure to dump code is not fatal
6264
if (fp) {
@@ -128,6 +130,9 @@ void register_jit_code(const void *code, size_t code_size,
128130
static std::mutex m;
129131
std::lock_guard<std::mutex> guard(m);
130132

133+
if (code && get_jit_dump()) {
134+
std::cout << "[ oneDNN ] register_jit_code: " << source_file_name << ", " << code_name << std::endl;
135+
}
131136
dump_jit_code(code, code_size, code_name);
132137
register_jit_code_vtune(code, code_size, code_name, source_file_name);
133138
register_jit_code_linux_perf(code, code_size, code_name, source_file_name);

0 commit comments

Comments
 (0)