NVIDIA/cutlass v4.6.1

CUTLASS 4.6.1

on GitHub

CuTe DSL

• Bug fixing and improvements
  • Fixed following issues:
    • #3243
    • #3359
    • #3365
    • #3312
  • Fixed a compilation failure on Thor with 12.9 wheel
  • Fixed a per regression of flash attention v2 on Ampere
  • Allow custom and multiple FFI call registration for Jax