benchmark

Add test to verify models and inputs are using the expected device

#443

Merged