TensorFlow-DirectML 1.15.5.dev210422
Pre-release
Pre-release
jstoecker
released this
23 Apr 06:04
·
137 commits
to directml
since this release
Preview build of tensorflow-directml built on April 22, 2021.
The Python packages are available as a PyPI release. To download the latest python package automatically, simply pip install tensorflow-directml.
Changes in dev210422
- Registers 32 new DirectML kernels. See the new Operator Roadmap for an overview of the operations DirectML supports.
- Updates to DirectML 1.5.0, which includes several shader performance improvements for training-related operators.
- Greatly improves threading efficiency of the DML execution backend, which improves performance in models that comprise large numbers of inexpensive operators.
- Resolves fatal errors when attempting to run TensorFlow-DirectML on older versions of Windows that do not support DirectML (falls back to CPU device).
- Forces integrated GPUs (UMA adapters) to grow their memory requirements dynamically instead of upfront like discrete adapters. This should resolve immediate device removal on some hardware.
- Merges upstream TensorFlow 1.15.5 fixes.
- Fixes several DirectML kernel bugs.