mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-14 01:18:26 -05:00
19 lines
519 B
Plaintext
19 lines
519 B
Plaintext
#pragma once
|
|
|
|
#include "../ops/ops.cuh"
|
|
#include "club.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 != cudaSuccess)
|
|
{
|
|
std::cerr << "CUDA Runtime Error at: " << file << ":" << line
|
|
<< std::endl;
|
|
std::cerr << cudaGetErrorString(err) << " " << func << std::endl;
|
|
//std::exit(EXIT_FAILURE);
|
|
}
|
|
} |