64da26ef5d
* yolov7 v0.1 * Update README.md * Update README.md * Update README.md * Update README.md * Update README.md * Update README.md * Update README.md * Update yololayer.h * Update yololayer.h * Delete yolov7.engine * Delete yolov7-tiny.engine * Delete yolov7/output directory * Delete dirent.h * Update README.md * Update README.md * Update README.md * Update README.md
19 lines
417 B
C++
19 lines
417 B
C++
#ifndef TRTX_CUDA_UTILS_H_
|
|
#define TRTX_CUDA_UTILS_H_
|
|
|
|
#include <cuda_runtime_api.h>
|
|
|
|
#ifndef CUDA_CHECK
|
|
#define CUDA_CHECK(callstr)\
|
|
{\
|
|
cudaError_t error_code = callstr;\
|
|
if (error_code != cudaSuccess) {\
|
|
std::cerr << "CUDA error " << error_code << " at " << __FILE__ << ":" << __LINE__;\
|
|
assert(0);\
|
|
}\
|
|
}
|
|
#endif // CUDA_CHECK
|
|
|
|
#endif // TRTX_CUDA_UTILS_H_
|
|
|