Skip to content

TensorFlow-DirectML 1.15.5.dev210422

Pre-release
Pre-release
Compare
Choose a tag to compare
@jstoecker jstoecker released this 23 Apr 06:04
· 137 commits to directml since this release
121f085

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.