Fix import for torch 2.0, 2.1 - guard typehint for "device_mesh" #36768
qubvel
marked this pull request as ready for review 1 year ago
qubvel
force pushed
from
1d0132b5
to
ed54e327
1 year ago
Fix device_mesh
dc8e7b33
qubvel
force pushed
from
ed54e327
to
dc8e7b33
1 year ago
Remove rebase leftover
adbdfcc4
qubvel
merged
cf8091c0
into main 1 year ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub