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