Skip to content

v0.3.1 - Fix missing CUDA extensions in wheel

Choose a tag to compare

@lyehe lyehe released this 02 Jan 05:01
· 26 commits to main since this release

Bug Fix

Fixed an issue where CUDA extensions were not included in the PyPI wheel.

What was wrong

  • The package_data pattern _cuda_* didn't match .pyd extension files
  • Result: wheels were built without CUDA extensions, causing import errors

Fix

  • Changed pattern to *.pyd and *.so to properly include all extensions
  • Now the fat wheel correctly bundles all 4 CUDA variants

Full Changelog: v0.3.0...v0.3.1