xla
0576fe95 - fix issue that torchbench uses xla as device instead of tpu (#6341)

Commit

2 years ago

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

References

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

Author

zpcore

zpcore

Parents

Loading