Intel® oneAPI Deep Neural Network Developer Guide and Reference
A newer version of this document is available. Customers should click here to go to the newest version.
Visible to Intel only — GUID: GUID-6B2F9A16-A2EF-4BAB-983C-ECF3B87B3AB9
Visible to Intel only — GUID: GUID-6B2F9A16-A2EF-4BAB-983C-ECF3B87B3AB9
Primitive Cache
Primitive creation time largely depends on the underlying implementation, for instance, oneDNN uses just-in-time compilation (JIT) to generate optimal code for some CPU and GPU implementations, which introduces overhead.
To mitigate primitive creation overhead, oneDNN provides the primitive cache which automatically caches created primitives to avoid repeating JIT compilation for the primitives with identical primitive parameters, attributes, underlying primitive implementations, etc. It can significantly reduce primitive creation overhead, especially when an application or a framework creates primitives for every instance of inference or iteration of training process.
The primitive cache is global hence a user does not have to maintain any persistent oneDNN resources to benefit from the primitive cache.
Managing Memory Consumption
The primitive cache has an upper limit for the number of primitives stored. Once capacity is exceeded, a primitive that was least recently used will be evicted from the cache. See the Run-time Controls section below for information on changing the cache capacity.
Profiling
Information about primitive cache hits and misses can be used for debug purposes. That information is part of the verbose output for verbose level 2 (Verbose Mode).
Build-time Controls
At build-time, support for this feature is controlled via cmake option ONEDNN_ENABLE_PRIMITIVE_CACHE.
CMake Option |
Supported values (defaults in bold) |
Description |
|---|---|---|
ONEDNN_ENABLE_PRIMITIVE_CACHE |
ON , OFF |
Enables primitive cache |
Run-time Controls
When the feature is enabled at build-time, the ONEDNN_PRIMITIVE_CACHE_CAPACITY environment variable can be used to change cache capacity or disable the cache.
Environment variable |
Value |
Description |
|---|---|---|
ONEDNN_PRIMITIVE_CACHE_CAPACITY |
<number> 0 |
Set cache capacity to <number> (default 1024 ) Disable primitive cache |
This feature can also be managed at run-time with the following functions:
The function setting takes precedence over the environment variable.