mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
a016150163
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/60543 Since now c10d is part of libtorch, it would also be nice if the sources lived all in one place. ghstack-source-id: 132306292 Test Plan: It builds Reviewed By: cbalioglu Differential Revision: D29062002 fbshipit-source-id: d9e1301e9d73e1643fa0f0119cd2d618f1ad52e6
20 lines
340 B
C++
20 lines
340 B
C++
#pragma once
|
|
|
|
#include <c10/cuda/CUDAStream.h>
|
|
|
|
namespace c10d {
|
|
namespace test {
|
|
|
|
#ifdef _WIN32
|
|
#define EXPORT_TEST_API __declspec(dllexport)
|
|
#else
|
|
#define EXPORT_TEST_API
|
|
#endif
|
|
|
|
EXPORT_TEST_API void cudaSleep(at::cuda::CUDAStream& stream, uint64_t clocks);
|
|
|
|
EXPORT_TEST_API int cudaNumDevices();
|
|
|
|
} // namespace test
|
|
} // namespace c10d
|