mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-05-04 14:38:10 +00:00
279300dc5f
* refactor compilation mechanism and unified multi hw * fix file path bug * add init.py to make pybind a module to avoid relative path error caused by softlink * delete duplicated micros * fix micros bug in gcc
19 lines
545 B
C++
19 lines
545 B
C++
#pragma once
|
|
|
|
#include <cuda.h>
|
|
#include <cuda_runtime.h>
|
|
|
|
#include <exception>
|
|
|
|
#define CUDA_CHECK(func) \
|
|
{ \
|
|
auto status = func; \
|
|
if (status != cudaSuccess) { \
|
|
throw std::runtime_error(cudaGetErrorString(status)); \
|
|
} \
|
|
}
|
|
|
|
#define HOST __host__
|
|
#define DEVICE __device__
|
|
#define HOSTDEVICE __host__ __device__
|