Summary:
As it uses `std::cerr`, which in turn results in compilation regression introduced by https://github.com/pytorch/pytorch/pull/61500
Fixes https://github.com/pytorch/pytorch/issues/63653
Pull Request resolved: https://github.com/pytorch/pytorch/pull/63656
Reviewed By: ejguan
Differential Revision:
D30455824
Pulled By: malfet
fbshipit-source-id:
29f316e7f7fd8e7dcbee2666e7a985f25bf56515
#include <limits>
#include <map>
+#include <iostream>
#include <c10/core/DeviceGuard.h>
#include <c10/util/irange.h>