benchmark
10d61e8e - Fix bench w/ iobinding; Remove cpu fallback (#113703)

Commit
2 years ago
Fix bench w/ iobinding; Remove cpu fallback (#113703) Summary: Summary - `TORCH_TO_NUMPY_DTYPE` was misplaced previously hence subclasses cannot access it. - Remove cpu fallback when benching onnx with gpu, expose gpu run failures properly. X-link: https://github.com/pytorch/pytorch/pull/113703 Approved by: https://github.com/thiagocrepaldi ghstack dependencies: #113404, #113697 Reviewed By: huydhn Differential Revision: D51444513 fbshipit-source-id: b934f438290ccdd8b443aa9b103b2b9e4dddf29d
Author
Parents
Loading