CPU ISA Hints¶
For performance reasons, extra hints may be provided to oneDNN which enable the just-in-time (JIT) code generation to prefer or avoid certain CPU ISA features. An example is disabling Zmm vector registers usage in order to take advantage of higher clock speed.
Build-time Controls¶
At build-time, support for this feature is controlled via cmake option ONEDNN_ENABLE_CPU_ISA_HINTS.
CMake Option |
Supported values (defaults in bold) |
Description |
|---|---|---|
ONEDNN_ENABLE_CPU_ISA_HINTS |
ON , OFF |
Enables CPU ISA hints |
This results in making oneDNN aware of ONEDNN_CPU_ISA_HINTS environment variable and corresponding setter utility namely, dnnl::set_cpu_isa_hints routine.
Run-time Controls¶
When the feature is enabled at build-time, the ONEDNN_CPU_ISA_HINTS environment variable can be used to specify an ISA-specific hint to enable oneDNN to dispatch an appropriate kernel.
Environment variable |
Value |
Description |
|---|---|---|
ONEDNN_CPU_ISA_HINTS |
NO_HINTS |
Use default configuration for ISA |
PREFER_YMM |
Prefer to use YMM registers for vector operations |
This feature can also be managed at run-time with the following functions:
dnnl::set_cpu_isa_hints function allows changing the CPU ISA hint at run-time. The limitation is that it is possible to set the value only once. In addition, it is advised to call this function before any other oneDNN API. This is because the first internal query of CPU ISA hints will disable the ability to change it. Once disabled, changing the CPU ISA hint will return an error.
dnnl::get_cpu_isa_hints function returns the currently used CPU ISA hints and by default poses no restrictions.
Function settings take precedence over environment variables. Moreover, if the hint is not applicable then it would be silently ignored. For instance with SSE41 there are no YMM registers and so hint to prefer YMM registers would be silently bypassed.