xla
fix issue that torchbench uses xla as device instead of tpu
#6341

Merged

fix issue that torchbench uses xla as device instead of tpu #6341

zpcore merged 4 commits into master from piz/bug_fix

fix issue that torchbench uses xla as device instead of tpu

16a9cf55

zpcore requested a review from

ysiraichi 2 years ago

zpcore requested a review from

miladm 2 years ago

nit update

a694479d

ysiraichi commented on 2024-01-21

consider device number for xla

047ea000

miladm approved these changes on 2024-01-22

Merge remote-tracking branch 'origin' into piz/bug_fix

13390fa7

zpcore force pushed from 3ab09810 to 13390fa7 2 years ago

zpcore merged 0576fe95 into master 2 years ago

zpcore deleted the piz/bug_fix branch 2 years ago

Reviewers

miladm

ysiraichi

Assignees

No one assigned

Labels

None yet

Milestone

No milestone