pytorch
19dda7c6 - Fallback to CPU when remote end does not have CUDA for profiling (#44967)

Commit
4 years ago
Fallback to CPU when remote end does not have CUDA for profiling (#44967) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/44967 When enabling profiler on server, if it is a different machine it may not have CUDA while caller does. In this case, we would crash but now we fallback to CPU and log a warning. ghstack-source-id: 112977906 Test Plan: CI Reviewed By: pritamdamania87 Differential Revision: D23790729 fbshipit-source-id: dc6eba172b7e666842d54553f52a6b9d5f0a5362
Author
Parents
Loading