9cfc7967dd
* Replace license text in headers with SPDX identifiers. * Remove specific license info from outdated readme.txt, instead leave details to the source files. * Add list of SPDX license identifiers used, and corresponding license texts. * Update copyright dates while we're at it. Ref D14069, T95597
53 lines
1.2 KiB
C++
53 lines
1.2 KiB
C++
/* SPDX-License-Identifier: Apache-2.0
|
|
* Copyright 2011-2022 Blender Foundation */
|
|
|
|
#pragma once
|
|
|
|
#ifdef WITH_CUDA
|
|
|
|
# ifdef WITH_CUDA_DYNLOAD
|
|
# include "cuew.h"
|
|
# else
|
|
# include <cuda.h>
|
|
# endif
|
|
|
|
CCL_NAMESPACE_BEGIN
|
|
|
|
class CUDADevice;
|
|
|
|
/* Utility to push/pop CUDA context. */
|
|
class CUDAContextScope {
|
|
public:
|
|
CUDAContextScope(CUDADevice *device);
|
|
~CUDAContextScope();
|
|
|
|
private:
|
|
CUDADevice *device;
|
|
};
|
|
|
|
/* Utility for checking return values of CUDA function calls. */
|
|
# define cuda_device_assert(cuda_device, stmt) \
|
|
{ \
|
|
CUresult result = stmt; \
|
|
if (result != CUDA_SUCCESS) { \
|
|
const char *name = cuewErrorString(result); \
|
|
cuda_device->set_error( \
|
|
string_printf("%s in %s (%s:%d)", name, #stmt, __FILE__, __LINE__)); \
|
|
} \
|
|
} \
|
|
(void)0
|
|
|
|
# define cuda_assert(stmt) cuda_device_assert(this, stmt)
|
|
|
|
# ifndef WITH_CUDA_DYNLOAD
|
|
/* Transparently implement some functions, so majority of the file does not need
|
|
* to worry about difference between dynamically loaded and linked CUDA at all. */
|
|
const char *cuewErrorString(CUresult result);
|
|
const char *cuewCompilerPath();
|
|
int cuewCompilerVersion();
|
|
# endif /* WITH_CUDA_DYNLOAD */
|
|
|
|
CCL_NAMESPACE_END
|
|
|
|
#endif /* WITH_CUDA */
|