accelerate
fix mlu device longTensor bugs
#2887
Merged

fix mlu device longTensor bugs #2887

muellerzr merged 34 commits into huggingface:main from huismiling:main
huismiling
huismiling Add Cambricon MLU accelerator support
bc5ccfb4
huismiling up mlu support for test
3ad38dcd
huismiling fix mlu device MULTI_MLU
be32c91c
huismiling Update src/accelerate/utils/imports.py
421c1424
huismiling up mlu for quality check
78cd1cb6
huismiling fix mlu device longTensor error
3abd0383
huismiling fix mlu device tensor dtype check
0542987f
huismiling fix mlu device send_to_device with torch dynamo error
e024276c
muellerzr Refactor AcceleratorState
a50b0d91
muellerzr Should be near complete now
ff628be7
muellerzr Last missing piece
a1aac83c
muellerzr Make my way to the acceleratorstate
31ea8cc5
muellerzr Include update to global var
47b60ca6
muellerzr Don't use global
2082a9a2
muellerzr gpu -> cuda
26c484e2
muellerzr Don't use update for dict, easier to read
5ac5d566
muellerzr Fix tests
2baa5c3b
muellerzr stash
d709f666
muellerzr Getting closer...
ac243159
muellerzr Needed to spawn at the very end after env was setup
16288982
muellerzr Explain set_device before deepspeed
6958e1bb
muellerzr Make docstring more accurate
2b9d339c
muellerzr Early return insteaD
194db939
muellerzr Delineat blocks
31201d3d
muellerzr Make prepare_backend return state + backend for clarity/less magic
eef1aa0b
huismiling Merge branch 'huggingface:main' into main
37d0edcd
huismiling Merge remote-tracking branch 'hf-acc/refactor-state'
0fc1df32
huismiling Merge branch 'huggingface:main' into main
b09003c6
huismiling merge from hf
92dc4bca
huismiling fix mlu longtensor.to() bugs.
124331ad
huismiling Merge branch 'huggingface:main' into main
36f35e87
huismiling Merge branch 'huggingface:main' into main
48d2c0c0
huismiling Merge branch 'huggingface:main' into main
900efd00
huismiling Merge branch 'huggingface:main' into main
b3a1aed2
SunMarc
SunMarc approved these changes on 2024-06-25
muellerzr
muellerzr commented on 2024-07-03
HuggingFaceDocBuilderDev
huismiling
muellerzr muellerzr merged fec1170e into main 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone