cuda_utils.h

#pragma once
#include <cstdlib>
#include <cuda_runtime.h>
#include <iostream>


// Macro kiểm tra lỗi CUDA chuẩn
#define CUDA_CHECK(call)                                                       \
  do {                                                                         \
    cudaError_t err = call;                                                    \
    if (err != cudaSuccess) {                                                  \
      std::cerr << "CUDA Error: " << cudaGetErrorString(err)                   \
                << " (err_num=" << err << ") at " << __FILE__ << ":"           \
                << __LINE__ << std::endl;                                      \
      exit(1);                                                                 \
    }                                                                          \
  } while (0)

// Hàm kiểm tra kernel launch (gọi sau mỗi kernel <<<...>>>)
inline void checkKernelError(const char *msg) {
  cudaError_t err = cudaGetLastError();
  if (err != cudaSuccess) {
    std::cerr << "Kernel Launch Error (" << msg
              << "): " << cudaGetErrorString(err) << std::endl;
    exit(1);
  }
}