Skip to content
This repository was archived by the owner on Apr 5, 2023. It is now read-only.

Ivan/mkl variable input size fix refactor#2

Open
mavrov wants to merge 5 commits into
ivan/mkl-variable-input-size-fixfrom
ivan/mkl-variable-input-size-fix-refactor
Open

Ivan/mkl variable input size fix refactor#2
mavrov wants to merge 5 commits into
ivan/mkl-variable-input-size-fixfrom
ivan/mkl-variable-input-size-fix-refactor

Conversation

@mavrov

@mavrov mavrov commented Nov 13, 2018

Copy link
Copy Markdown

No description provided.

When MKL is used with variable size tensors the MKL primitive reuse map does not have a good hit rate and grows until system goes OOM.
Allocated MKL primitives never got deallocated.
Introducing a new backwards compatible environment variable TF_MKL_REUSE_PRIMITIVE_MEMORY to allow primitive reuse disabling.
@mavrov mavrov self-assigned this Nov 13, 2018
@mavrov
mavrov force-pushed the ivan/mkl-variable-input-size-fix branch from 04a2d14 to 3015347 Compare March 6, 2019 14:19
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant