mirror of
https://github.com/ziglang/zig.git
synced 2024-11-30 09:02:32 +00:00
65 lines
1.8 KiB
C
Vendored
65 lines
1.8 KiB
C
Vendored
/*===---- __clang_hip_runtime_wrapper.h - HIP runtime support ---------------===
|
|
*
|
|
* Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
* See https://llvm.org/LICENSE.txt for license information.
|
|
* SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
*
|
|
*===-----------------------------------------------------------------------===
|
|
*/
|
|
|
|
/*
|
|
* WARNING: This header is intended to be directly -include'd by
|
|
* the compiler and is not supposed to be included by users.
|
|
*
|
|
*/
|
|
|
|
#ifndef __CLANG_HIP_RUNTIME_WRAPPER_H__
|
|
#define __CLANG_HIP_RUNTIME_WRAPPER_H__
|
|
|
|
#if __HIP__
|
|
|
|
#include <cmath>
|
|
#include <cstdlib>
|
|
#include <stdlib.h>
|
|
|
|
#define __host__ __attribute__((host))
|
|
#define __device__ __attribute__((device))
|
|
#define __global__ __attribute__((global))
|
|
#define __shared__ __attribute__((shared))
|
|
#define __constant__ __attribute__((constant))
|
|
|
|
#if __HIP_ENABLE_DEVICE_MALLOC__
|
|
extern "C" __device__ void *__hip_malloc(size_t __size);
|
|
extern "C" __device__ void *__hip_free(void *__ptr);
|
|
static inline __device__ void *malloc(size_t __size) {
|
|
return __hip_malloc(__size);
|
|
}
|
|
static inline __device__ void *free(void *__ptr) { return __hip_free(__ptr); }
|
|
#else
|
|
static inline __device__ void *malloc(size_t __size) {
|
|
__builtin_trap();
|
|
return nullptr;
|
|
}
|
|
static inline __device__ void *free(void *__ptr) {
|
|
__builtin_trap();
|
|
return nullptr;
|
|
}
|
|
#endif
|
|
|
|
#include <__clang_hip_libdevice_declares.h>
|
|
#include <__clang_hip_math.h>
|
|
|
|
#if !_OPENMP || __HIP_ENABLE_CUDA_WRAPPER_FOR_OPENMP__
|
|
#include <__clang_cuda_math_forward_declares.h>
|
|
#include <__clang_cuda_complex_builtins.h>
|
|
|
|
#include <algorithm>
|
|
#include <complex>
|
|
#include <new>
|
|
#endif // !_OPENMP || __HIP_ENABLE_CUDA_WRAPPER_FOR_OPENMP__
|
|
|
|
#define __CLANG_HIP_RUNTIME_WRAPPER_INCLUDED__ 1
|
|
|
|
#endif // __HIP__
|
|
#endif // __CLANG_HIP_RUNTIME_WRAPPER_H__
|