"vllm_omni/distributed/omni_connectors/utils/logging.py" did not exist on "356077823ea8569ff15218e51228c1b3d50792a9"
infinirt_cuda.cuh 356 Bytes
Newer Older
1
2
#ifndef __INFINIRT_CUDA_H__
#define __INFINIRT_CUDA_H__
PanZezhong's avatar
PanZezhong committed
3
#include "../infinirt_impl.h"
4
5

namespace infinirt::cuda {
6
#if defined(ENABLE_NVIDIA_API) || defined(ENABLE_ILUVATAR_API) || defined(ENABLE_QY_API) || defined(ENABLE_HYGON_API)
7
8
9
10
11
12
13
INFINIRT_DEVICE_API_IMPL
#else
INFINIRT_DEVICE_API_NOOP
#endif
} // namespace infinirt::cuda

#endif // __INFINIRT_CUDA_H__