Skip to content

torch-directml 0.2.4.dev240815

Pre-release
Pre-release
Compare
Choose a tag to compare
@joshjkim joshjkim released this 19 Aug 20:59
· 15 commits to master since this release
torch-directml-0.2.4.dev240815
07377a6
This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Verified
Learn about vigilant mode.

Preview build of torch-directml built on August 15, 2024.

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

What’s New

  • Operator support for avg_pool3d, upsample_bicubic2d
  • Support for device_count in torch PrivateUse1 backend

Bug Fixes

  • Fix for Whisper sample not running in fp16 with DirectML scaled dot product attention #598
  • Issues addressed with operator support and device_count inPrivateUse1 #596 #609
Assets 12
Loading