accelerate
5d4a3beb - [tests] use `torch_device` instead of `0` for device check (#2861)

Commit

1 year ago

[tests] use `torch_device` instead of `0` for device check (#2861) * bug fix * fix one more case * add more cases * refine

References

Author

faaany

Parents