benchmark
c8f21e99 - Add test to verify models and inputs are using the expected device (#443)

Commit
4 years ago
Add test to verify models and inputs are using the expected device (#443) * Add test to verify models and inputs are using the expected device The device attribute needs to be set in three locations: the BenchmarkModel, the nn.Module model, and the model inputs. This test will verify that the tensors and inputs are on the expected device set in BenchmarkModel. Also, fix a few benchmark bugs where the device is incorrectly set. * Fix a bug in tts_angular benchmark not allocating inputs on expected device * Minor improvements to check_device Change Exception to RuntimeError. Update check inputs to a recursive implementation. * Fix small mistake for inputs tensor
Author
Parents
Loading