Skip to content

GitLab

"mmdet3d/vscode:/vscode.git/clone" did not exist on "2bd99e8f4ec4874960c63a7aecff371ba9636c3e"
Switch branch/tag
Find file Normal viewHistoryPermalink
config.h.in 292 Bytes
Newer Older
Chao Liu's avatar
hip build  
Chao Liu committed
1
2
3
4
5
6 
#pragma once
#cmakedefine01 DEVICE_BACKEND_HIP
#cmakedefine01 DEVICE_BACKEND_CUDA

#if DEVICE_BACKEND_HIP
#include "hip/hip_runtime.h"
Chao Liu's avatar
hip build  
Chao Liu committed
7 
#include "hip/hip_fp16.h"
Chao Liu's avatar
hip build  
Chao Liu committed
8
9 
#elif DEVICE_BACKEND_CUDA
#include "cuda_runtime.h"
Chao Liu's avatar
hip build  
Chao Liu committed
10 
#include "cuda_fp16.h"
Chao Liu's avatar
hip build  
Chao Liu committed
11
12
13 
#include "nvToolsExt.h"
#include "helper_cuda.h"
#endif