mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-07-17 23:32:07 +08:00
firestar5683
18 lines
496 B
Plaintext
18 lines
496 B
Plaintext
#pragma once
|
|
|
|
#include "../ops/ops.cuh"
|
|
#include <iostream>
|
|
|
|
#define CHECK_CUDA_ERROR(val) check((val), #val, __FILE__, __LINE__)
|
|
template <typename T>
|
|
void check(T err, char const* const func, char const* const file,
|
|
int const line)
|
|
{
|
|
if (err != hipSuccess)
|
|
{
|
|
std::cerr << "HIP Runtime Error at: " << file << ":" << line
|
|
<< std::endl;
|
|
std::cerr << hipGetErrorString(err) << " " << func << std::endl;
|
|
//std::exit(EXIT_FAILURE);
|
|
}
|
|
} |