nvrtc_jit.h

Go to the documentation of this file.

#pragma once
 
#include <string>
 
namespace fz {
namespace fused {
 
bool nvrtcAvailable();
 
void* nvrtcGetKernel(const std::string& src, const char* entry);
 
[[noreturn]] void cuThrow(int result, const char* prefix, const char* what);
 
#define FZ_CU_CHECK(call, prefix) \
    do { \
        CUresult _fz_cu_r = (call); \
        if (_fz_cu_r != CUDA_SUCCESS) ::fz::fused::cuThrow((int)_fz_cu_r, prefix, #call); \
    } while (0)
 
} // namespace fused
} // namespace fz