Skip to content

Commit ad92b6b

Browse files
luweizhou2016azhai219
authored andcommitted
[FORK][FEAUTRE] Add log to jit dump code
1 parent 0af0320 commit ad92b6b

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/cpu/jit_utils/jit_utils.cpp

+5-2
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
#include "common/verbose.hpp"
@@ -60,7 +61,7 @@ void dump_jit_code(const void *code, size_t code_size, const char *code_name) {
6061
// TODO (Roma): support prefix for code / linux perf dumps
6162
snprintf(fname, MAX_FNAME_LEN, DUMP_BASE_FNAME "%s" DUMP_EXT_FNAME,
6263
code_name);
63-
64+
std::cout << "[ oneDNN ] dump_jit_code: " << fname << std::endl;
6465
FILE *fp = fopen(fname, "wb+");
6566
// Failure to dump code is not fatal
6667
if (fp) {
@@ -137,7 +138,9 @@ void register_jit_code(const void *code, size_t code_size,
137138
char unique_code_name[MAX_CODENAME_LEN + 1];
138139
snprintf(unique_code_name, MAX_CODENAME_LEN, "%s.%d", code_name,
139140
unique_id++);
140-
141+
if (code && get_jit_dump()) {
142+
std::cout << "[ oneDNN ] register_jit_code: " << unique_code_name << ", " << code_name << std::endl;
143+
}
141144
dump_jit_code(code, code_size, unique_code_name);
142145
// VTune Profiler does not need a unique name, because it uses
143146
// unique method_id

0 commit comments

Comments
 (0)