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
|