pytorch/c10/cuda/CUDAMiscFunctions.h
Natalia Gimelshein 6284d2a82b wrap cudaStreamSynchronize calls (#61889)
Summary:
This is a first step towards creating context manager that errors out on synchronizing calls.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/61889

Reviewed By: albanD

Differential Revision: D29805280

Pulled By: ngimel

fbshipit-source-id: b66400fbe0941b7daa51e6b30abe27b9cccd4e8a
2021-07-21 19:30:52 -07:00

11 lines
254 B
C++

#pragma once
// this file is to avoid circular dependency between CUDAFunctions.h and
// CUDAExceptions.h
#include <c10/cuda/CUDAMacros.h>
namespace c10 {
namespace cuda {
C10_CUDA_API const char* get_cuda_check_suffix() noexcept;
}
} // namespace c10