Commit Graph

2 Commits

Author SHA1 Message Date
Nicolas Macchioni
ee3e89190a refactor benchmarking to use dynamo_timed (#144315)
use dynamo_timed for all our wrapped calls, instead of our custom timer

Pull Request resolved: https://github.com/pytorch/pytorch/pull/144315
Approved by: https://github.com/eellison
2025-01-18 02:35:00 +00:00
Nicolas Macchioni
af664882dd Safely infer device type + docstrings + tests (#133668)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/133668
Approved by: https://github.com/eellison
2024-08-21 05:27:31 +00:00