cuml_api.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2019-2024, NVIDIA CORPORATION.
 * SPDX-License-Identifier: Apache-2.0
 */
 
#pragma once
 
#include <cuda_runtime_api.h>
 
#include <stddef.h>
 
// Block inclusion of this header when compiling libcuml++.so. If this error is
// shown during compilation, there is an issue with how the `#include` have
// been set up. To debug the issue, run `./build.sh cppdocs` and open the page
// 'cpp/build/html/cuml__api_8h.html' in a browser. This will show which files
// directly and indirectly include this file. Only files ending in '*_api' or
// 'cumlHandle' should include this header.
#ifdef CUML_CPP_API
#error "This header is only for the C-API and should not be included from the C++ API."
#endif
 
#ifdef __cplusplus
extern "C" {
#endif
 
typedef int cumlHandle_t;
 
typedef enum cumlError_t { CUML_SUCCESS, CUML_ERROR_UNKNOWN, CUML_INVALID_HANDLE } cumlError_t;
 
typedef cudaError_t (*cuml_allocate)(void** p, size_t n, cudaStream_t stream);
typedef cudaError_t (*cuml_deallocate)(void* p, size_t n, cudaStream_t stream);
 
const char* cumlGetErrorString(cumlError_t error);
 
cumlError_t cumlCreate(cumlHandle_t* handle, cudaStream_t stream);
 
cumlError_t cumlGetStream(cumlHandle_t handle, cudaStream_t* stream);
 
cumlError_t cumlSetDeviceAllocator(cumlHandle_t handle,
                                   cuml_allocate allocate_fn,
                                   cuml_deallocate deallocate_fn);
cumlError_t cumlSetHostAllocator(cumlHandle_t handle,
                                 cuml_allocate allocate_fn,
                                 cuml_deallocate deallocate_fn);
 
cumlError_t cumlDestroy(cumlHandle_t handle);
 
#ifdef __cplusplus
}
#endif