This is the release note of v9.4.0. See here for the complete list of solved issues and merged PRs.

We are running a Gitter chat for general discussions and quick questions. Feel free to join the channel to talk with developers and users!

Highlights

Compile with SASS (CUBIN) for CUDA versions >= 11.1 (#5097)

Changes NVRTC compile process to produce SASS (CUBIN files) instead of PTX so that kernels compiled with a new CUDA Toolkit version can be run with earlier CUDA Drivers. Check the CUDA Compatibility Guide and NVRTC Documentation for detailed information. We believe most users will not be affected by this change, but you can revert to the previous behavior by setting CUPY_COMPILE_WITH_PTX=1 environment variable just in case.

Support for AMD ROCm 4.3

Support for ROCm 4.3 has been added in the latest release and binary wheels are provided as well. Note that there is currently an issue with ROCm 4.3 that prevents it from running in several environments. The current workaround is to set the LLVM_PATH variable to the llvm folder included in ROCm 4.3 installation (e.g., export LLVM_PATH=/opt/rocm-4.3/llvm).

Changes

Enhancements

• Compile with SASS for CUDA versions >= 11.1 (#5611)
• Allow to compile using PTX with an envvar (#5634)
• Add ncclAvg and ncclBfloat16 for NCCL (#5656)
• Fix version check for new ROCm version definition (#5661)
• Rest of version check fix for new ROCm version definition (#5670)

Bug Fixes

• Fix FFT convolve for shapes containing 1 (#5613)
• Fix the RTC call path for HIP (#5620)
• Fix compute capability check (#5646)
• Fix squareness checks (#5652)
• Fix unique for empty array (#5658)

Code Fixes

• Fix kernel names to be consistent (#5625)
• Remove unnecessary comments (#5635)

Documentation

• Update Sphinx to 4.1.2 (#5616)
• __array_function__ feature by default (#5653)
• Support ROCm v4.3 in document (#5674)

Tests

• Increase test timeout (#5615)
• Increase timeout for CUDA 11.4 tests (#5617)
• Add CI for ROCm 4.3 (#5632)
• Reload GPG key for ROCm 4.2 test (#5637)
• Fix cubic for_all_dtypes_combination tests (#5639)
• Add a workaround for ROCm 4.3.0 for testing (#5663)
• Fix skipTest in test_decomp_lu (#5672)

Others

• Bump version to v9.4.0 (#5680)

Contributors

The CuPy Team would like to thank all those who contributed to this release!

@grlee77 @leofang @yashasvimisra2798