[MLU] Fix FA2 check error, remove deepspeed-mlu deps. #36159
ArthurZucker
merged
50 commits into
huggingface:main
from
huismiling:main
add Cambricon MLUs support
2d367365
fix mlu device rng state
3ac7123e
up for quality check
51ec085e
up mlu to support fp16
db0fbe63
fix mlu device dependency error
fe39b569
fix mlu device dependency error
c43b68e7
enable mlu device for bf16
bd4d3f9e
fix mlu device memory tracker
988b2d65
merge from hf main
e33678f6
Merge remote-tracking branch 'hf-transformers/main'
06006014
Merge branch 'huggingface:main' into main
432febfe
Merge branch 'huggingface:main' into main
44323c3f
Cambricon support SDPA and flash_attn
6adf3d9e
Merge branch 'huggingface:main' into main
9126fd5b
Merge branch 'huggingface:main' into main
7dc5505d
Merge branch 'huggingface:main' into main
e5cab7f4
Merge branch 'huggingface:main' into main
596174ed
Merge branch 'huggingface:main' into main
372c0289
MLU devices : Checks if `mlu` is available via an `cndev-based` check…
18aa1bb3
Merge branch 'main' into main
c2ce6065
Merge branch 'main' into main
6f9cca99
Merge branch 'main' into main
77701863
Merge branch 'main' into main
582a3b88
Merge branch 'main' into main
f28c9171
Merge branch 'main' into main
586329e8
Merge branch 'huggingface:main' into main
6584f799
Merge branch 'huggingface:main' into main
4a1e279e
Merge branch 'huggingface:main' into main
bdcaccbe
Fix mlu FA2 check. Remove deepspeed-mlu check. add mlu tests support.
f0006ba8
Merge branch 'huggingface:main' into main
50c9020e
fix testing errors.
7bb2fabc
Merge branch 'main' of github.com:huismiling/transformers
647788ec
Merge branch 'huggingface:main' into main
a13ca7ca
Merge branch 'huggingface:main' into main
ec1f1f15
Merge branch 'main' into main
3b1b4bca
Merge branch 'main' into main
3a8865ff
Merge branch 'hf/main' into main
3d7fd499
Merge pull request #3 from huggingface/main
4eb8a118
Merge branch 'main' into main
7185ecfd
Merge branch 'main' into main
c95a233c
fix get_device_count error.
92e9cce4
Merge branch 'main' into main
604f799b
fix mlu testing utils.
b501f181
fix code quality and style.
7ce78d08
Merge branch 'main' into main
9c3575d8
SunMarc
approved these changes
on 2025-03-25
switch to @require_torch_multi_accelerator
a60fa558
Merge branch 'main' of github.com:huismiling/transformers
b7df58ff
Merge branch 'main' into main
3d870fd2
Merge branch 'main' of github.com:huismiling/transformers
25cd6cd5
Merge branch 'main' into main
4efd7465
ArthurZucker
approved these changes
on 2025-03-31
Login to write a write a comment.
Login via GitHub