Jetson에서 실행 가능한 CUDA 그래프 인스턴스를 정리한다.
그래프 정의와 별도로 만든 실행 인스턴스를 먼저 정리한다. 예제는 빈 그래프를 사용하며 실제 커널을 실행하지 않는다.
#include <cuda_runtime.h>
#include <cstdio>
#include <cstdlib>
#define CHECK(call) do { \
cudaError_t cuda_status_ = (call); \
if (cuda_status_ != cudaSuccess) { \
fprintf(stderr, "%s\n", cudaGetErrorString(cuda_status_)); \
exit(1); \
} \
} while (0)
int main() {
cudaGraph_t graph;
cudaGraphExec_t exec;
CHECK(cudaGraphCreate(&graph, 0));
CHECK(cudaGraphInstantiate(&exec, graph, 0));
CHECK(cudaGraphExecDestroy(exec));
CHECK(cudaGraphDestroy(graph));
return 0;
}CHECK는 이 예제에 정의한 오류 검사 매크로다. CUDA 호출이 실패하면 설명을 출력하고 종료하므로 실패한 메모리나 핸들을 계속 사용하지 않는다.
Jetson에서 빌드하기
nvcc -std=c++17 main.cu -o main
./main
답글 달기