accelerate

[tests] use `torch_device` instead of `0` for device check

#2861

Merged