How to print out Memory Descriptors and other stuff?

[Devjiu]

How to print out Memory Descriptors and other stuff?

[yehudaorel]

Hi @Devjiu, as for the first question, oneDNN provides API execution tracing with primitive information via the environment variable ONEDNN_VERBOSE. Please refer to the verbose mode documentation for more detail.

As for the second question, indeed you can use set_data_handle() for example in examples/primitives/convolution.cpp you can replace the the following lines:

write_to_dnnl_memory(src_data.data(), user_src_mem);
write_to_dnnl_memory(weights_data.data(), user_weights_mem);
write_to_dnnl_memory(bias_data.data(), user_bias_mem);

with

user_src_mem.set_data_handle(src_data.data());
user_weights_mem.set_data_handle(weights_data.data());
user_bias_mem.set_data_handle(bias_data.data());

Another option is to pass data during memory creation:
auto user_src_mem = memory({src_dims, dt::f32, tag::nchw}, engine, src_data.data());

Hope this helps, let us know if you run to any other issues!

[Devjiu]

@yehudaorel
Thanks, for quick response)
I was sure, that already set this variable - but there left dnnl_set_verbose(0); in my code. Thanks for the hint.
Setting data pointer in c-tor also helped, much better.

Sorry for noob questions, my questions are resolved.