mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-14 02:02:46 +00:00
16 lines
319 B
C
16 lines
319 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
|
|
|
|
using index_t = uint32_t;
|