/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
extensions/csrc/kernel/cuda/utils/micros.h
18 строк
545 B
傅剑寒
[Inference/Refactor] Refactor compilation mechanism and unified multi hw (#5613)
24 апр 2024, 09:17
Не верифицирован
24 апр 2024, 09:17
279300d
Код
Авторство
О чём код?
#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__