mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
d408324350
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/15316 This starts cleaning up the files in c10 according to the module structure we decided on. Move to c10/util: - Half.h, Half-inl.h, Half.cpp, bitcasts.h Move to c10/core: - Device.h, Device.cpp - DeviceType.h, DeviceType.cpp i-am-not-moving-c2-to-c10 Reviewed By: dzhulgakov Differential Revision: D13498493 fbshipit-source-id: dfcf1c490474a12ab950c72ca686b8ad86428f63
13 lines
358 B
C++
13 lines
358 B
C++
#include <c10/core/Stream.h>
|
|
|
|
namespace c10 {
|
|
|
|
// Not very parseable, but I don't know a good compact syntax for streams.
|
|
// Feel free to change this into something more compact if needed.
|
|
std::ostream& operator<<(std::ostream& stream, const Stream& s) {
|
|
stream << "stream " << s.id() << " on device " << s.device();
|
|
return stream;
|
|
}
|
|
|
|
} // namespace c10
|