pytorch
4c99351d - [AMD] Remove num_gpu check for remote execution (#34318)

Commit
4 years ago
[AMD] Remove num_gpu check for remote execution (#34318) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/34318 Stop checking whether we have AMD GPU devices on the host, because we may be constructing a net on a machine without GPU, and run the net on another one with GPU Reviewed By: ajauhri Differential Revision: D20269562 fbshipit-source-id: 1f561086cacdcead3ce7c03c2d02c25336c8b11a
Author
Parents
Loading