Cyrus LeungandMaxime Fournioux d848800e88 [Misc] Move print_*_once from utils to logger (#11298) ...

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
Signed-off-by: Maxime Fournioux <55544262+mfournioux@users.noreply.github.com>
Co-authored-by: Maxime Fournioux <55544262+mfournioux@users.noreply.github.com>

2025-01-09 12:48:12 +08:00

..

__init__.py

[Core] Refactor Attention Take 2 (#3462)

2024-03-25 04:39:33 +00:00

abstract.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00

blocksparse_attn.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00

flash_attn.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00

flashinfer.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00

hpu_attn.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00

ipex_attn.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00

openvino.py

[Model][OpenVINO] Fix regressions from #8346 (#10045)

2024-11-06 04:19:15 +00:00

pallas.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00

placeholder_attn.py

[Doc] Update docs to refer to pooling models (#11093)

2024-12-11 13:36:27 +00:00

rocm_flash_attn.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00

torch_sdpa.py

[Misc] Move print_*_once from utils to logger (#11298)

2025-01-09 12:48:12 +08:00

utils.py

[torch.compile] support all attention backends (#10558)

2024-11-22 14:04:42 -08:00

xformers.py

[Misc] Move print_*_once from utils to logger (#11298)

2025-01-09 12:48:12 +08:00