mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-20 21:09:08 +00:00
14 lines
292 B
C
14 lines
292 B
C
#pragma once
|
|
#cmakedefine01 DEVICE_BACKEND_HIP
|
|
#cmakedefine01 DEVICE_BACKEND_CUDA
|
|
|
|
#if DEVICE_BACKEND_HIP
|
|
#include "hip/hip_runtime.h"
|
|
#include "hip/hip_fp16.h"
|
|
#elif DEVICE_BACKEND_CUDA
|
|
#include "cuda_runtime.h"
|
|
#include "cuda_fp16.h"
|
|
#include "nvToolsExt.h"
|
|
#include "helper_cuda.h"
|
|
#endif
|