Skip to content

TensorFlow-DirectML 1.15.5
Compare
Choose a tag to compare
@jstoecker jstoecker released this 04 Sep 01:36
v1.15.5
180b464
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

Build of tensorflow-directml built on September 3, 2021.

This marks the first generally available (non-preview) release of TensorFlow-DirectML, which is based on the 1.15.5 version of TensorFlow. See our roadmap for more details on our plans for releases based on TensorFlow 1.x and 2.x.

The Python packages are available as a PyPI release. To download the latest python package automatically, simply pip install tensorflow-directml.

Changes in 1.15.5

  • Registers 7 new DirectML kernels. See the new Operator Roadmap for an overview of the operations DirectML supports.
  • Upgrades to DirectML 1.7.0, which includes several shader performance improvements for training-related operators.
  • Significantly improves the DirectML memory allocator, which now supports tiled allocations to improve the utilization of dedicated video memory.
  • Improves handling of INT64 tensors in several kernels.
  • Removes dependency on WSL2 interoperability for Linux builds.
  • D3D12, DXGI, and DXCore are linked at run time instead of load time; the CPU device should be usable even if running in a pure Linux environment.
  • Modifies the CI/release pipelines to produce manylinux2010-compliant wheels that run on even more glibc-based WSL2 distros.
  • Fixes several DirectML kernel bugs.
Assets 4