xla
Automatically detect CUDA if it isn't set and default to upstream torch.cuda number of devices
#6605

Merged

Automatically detect CUDA if it isn't set and default to upstream torch.cuda number of devices #6605

changm merged 3 commits into master from changm/gpu

Automatically detect CUDA if it isn't set and default to a single GPU…

4a898343

changm requested a review from

JackCaoG 1 year ago

changm assigned

changm 1 year ago

JackCaoG requested a review from

will-cromar 1 year ago

JackCaoG requested a review from

vanbasten23 1 year ago

will-cromar approved these changes on 2024-02-23

vanbasten23 commented on 2024-02-23

vanbasten23 commented on 2024-02-24

Use upstream torch number of GPU Devices

2e256709

Fix string bug

5fcd42c0

changm changed the title ~~Automatically detect CUDA if it isn't set and default to a single GPU device~~ Automatically detect CUDA if it isn't set and default to upstream torch.cuda number of devices 1 year ago

changm requested a review from

vanbasten23 1 year ago

vanbasten23 commented on 2024-02-26

vanbasten23 approved these changes on 2024-02-26

changm merged cb4983e9 into master 1 year ago

changm deleted the changm/gpu branch 1 year ago

Reviewers

vanbasten23

will-cromar

JackCaoG

Assignees

changm

Labels

None yet

Milestone

No milestone

xla Automatically detect CUDA if it isn't set and default to upstream torch.cuda number of devices #6605 Merged

Automatically detect CUDA if it isn't set and default to upstream torch.cuda number of devices #6605

xla
Automatically detect CUDA if it isn't set and default to upstream torch.cuda number of devices
#6605

Merged