Searched refs:C10_CUDA_API (Results 1 – 11 of 11) sorted by relevance
23 C10_CUDA_API DeviceIndex device_count() noexcept;26 C10_CUDA_API DeviceIndex device_count_ensure_non_zero();28 C10_CUDA_API DeviceIndex current_device();30 C10_CUDA_API void set_device(DeviceIndex device);32 C10_CUDA_API void device_synchronize();34 C10_CUDA_API void warn_or_error_on_sync();37 C10_CUDA_API cudaError_t GetDeviceCount(int* dev_count);39 C10_CUDA_API cudaError_t GetDevice(DeviceIndex* device);41 C10_CUDA_API cudaError_t SetDevice(DeviceIndex device);43 C10_CUDA_API cudaError_t MaybeSetDevice(DeviceIndex device);[all …]
10 C10_CUDA_API const char* get_cuda_check_suffix() noexcept;11 C10_CUDA_API std::mutex* getFreeMutex();
25 class C10_CUDA_API FreeMemoryCallback {299 C10_CUDA_API extern std::atomic<CUDAAllocator*> allocator;461 struct C10_CUDA_API MemPool {479 struct C10_CUDA_API MemPoolContext {
36 #define C10_CUDA_API C10_CUDA_EXPORT macro38 #define C10_CUDA_API C10_CUDA_IMPORT macro
22 class C10_CUDA_API CUDAError : public c10::Error {93 C10_CUDA_API void c10_cuda_check_implementation(
16 class C10_CUDA_API CUDAAllocatorConfig {122 C10_CUDA_API void setAllocatorSettings(const std::string& env);
20 struct C10_CUDA_API CUDAStreamCaptureModeGuard {
84 class C10_CUDA_API CUDAKernelLaunchRegistry {
60 class C10_CUDA_API CUDAStream {
179 C10_CUDA_API void setHasPrimaryContext(bool (*func)(DeviceIndex)) { in setHasPrimaryContext()
7 C10_CUDA_API int c10_cuda_test();