Commit e15ae4e
committed
fix(cute_dsl/attention): use string literals for fence_proxy (cu13 wheel compat)
Fixes #3071.
`cute.arch.fence_proxy` is documented as taking string literals
(``"async.shared"``, ``"cta"``); the wrapper does ``ProxyKind.from_str`` /
``SharedSpace.from_str`` internally. The enum re-exports
``cute.arch.ProxyKind`` and ``cute.arch.SharedSpace`` are gated upstream
behind ``cutlass_dsl.target_version(exact_version="12.9")`` and so are
absent from the cu13 wheel that the ``flashinfer-ci-cu130`` docker image
uses, producing:
AttributeError: module 'cutlass.cute.arch' has no attribute 'ProxyKind'
at correction.py:267 and softmax.py:448. On cu12 wheels the enum form
still works but emits ``DeprecationWarning: Passing enum member directly
to SharedSpace.from_str() is deprecated. Please use string literals
instead`` (visible in our local test runs).
Switching the two call sites to the string form fixes both the cu13
breakage and the cu12 deprecation warning, and is the documented stable
API on every flashinfer-supported cutlass-dsl 4.4.x wheel.
Verified: python -m pytest tests/attention/test_modular_fmha_prefill.py \
tests/attention/test_cute_dsl_mla_decode.py
→ 456 passed, 20 skipped (no SharedSpace/ProxyKind warnings emitted).
Made-with: Cursor1 parent 6e67879 commit e15ae4e
2 files changed
+2
-8
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
263 | 263 | | |
264 | 264 | | |
265 | 265 | | |
266 | | - | |
267 | | - | |
268 | | - | |
269 | | - | |
| 266 | + | |
270 | 267 | | |
271 | 268 | | |
272 | 269 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
444 | 444 | | |
445 | 445 | | |
446 | 446 | | |
447 | | - | |
448 | | - | |
449 | | - | |
450 | | - | |
| 447 | + | |
451 | 448 | | |
452 | 449 | | |
453 | 450 | | |
| |||
0 commit comments