Information regarding threading backend in oneDNN

[vineel96]

Hello, I have doubts regarding threading backend of oneDNN:

1. Is threading backend is common for all oneDNN algorithms/kernels or it is dependent on operator kernel or algorithm?
2. If threading backend is different or dependent on operator kernel, can you provide information regarding file location for the same implementations?
3. Is threading backend dependent on architecture i.e intel and arm it will be different?
4. Also can you provide the threading backend implementation files location in oneDNN?

[shu1chen]

Hello @vineel96, the threading backend is common for all oneDNN algorithms/kernels and is the same for x86_64/aarch64 architecture. The source code of threading implementation is in oneDNN/src/common/dnnl_thread.hpp. For example, in aarch64/brgemm_matmul.cpp, it calls the parallel function from dnnl_thread.hpp for threading.