diff --git a/flash_attn/cute/pyproject.toml b/flash_attn/cute/pyproject.toml index 619ae2c5db9..1503556c122 100644 --- a/flash_attn/cute/pyproject.toml +++ b/flash_attn/cute/pyproject.toml @@ -22,7 +22,7 @@ classifiers = [ ] dependencies = [ - "nvidia-cutlass-dsl>=4.3.4,<4.4.0", + "nvidia-cutlass-dsl>=4.3.5,<4.4.0", "torch", "einops", "typing_extensions",