Cyrus Leung 9fb52e523a [V1] Support any head size for FlexAttention backend (#20467) ...

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>

2025-07-06 09:54:36 -07:00

..

__init__.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

cpu.py

[V0 deprecation] Remove V0 CPU/XPU/TPU backends (#20412)

2025-07-06 08:48:13 -07:00

cuda.py

[V1] Support any head size for FlexAttention backend (#20467)

2025-07-06 09:54:36 -07:00

hpu.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

interface.py

[Platform] Add custom default max tokens (#18557)

2025-07-04 10:50:17 +08:00

neuron.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

rocm.py

[UT][intel GPU] use current_platform instead of device hardcode in v1 tests (#20169)

2025-07-02 09:06:04 +08:00

tpu.py

[V0 deprecation] Remove V0 CPU/XPU/TPU backends (#20412)

2025-07-06 08:48:13 -07:00

xpu.py

[V0 deprecation] Remove V0 CPU/XPU/TPU backends (#20412)

2025-07-06 08:48:13 -07:00