From 3b4430906dc56732c2bd56d95a5151c2e3496632 Mon Sep 17 00:00:00 2001 From: li-xl <1905692338@qq.com> Date: Mon, 30 Nov 2020 16:42:46 +0800 Subject: [PATCH] add peek_logged --- extern/cuda/src/helper_cuda.cc | 1 - src/executor.h | 4 ++++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/extern/cuda/src/helper_cuda.cc b/extern/cuda/src/helper_cuda.cc index bd715d5a..8c8733f4 100644 --- a/extern/cuda/src/helper_cuda.cc +++ b/extern/cuda/src/helper_cuda.cc @@ -15,7 +15,6 @@ #include #include "helper_cuda.h" - #ifdef _CUFFT_H_ // cuFFT API errors const char *_cudaGetErrorEnum(cufftResult error) { diff --git a/src/executor.h b/src/executor.h index 6f93cd17..3c2801ac 100644 --- a/src/executor.h +++ b/src/executor.h @@ -7,6 +7,10 @@ #pragma once #include "common.h" #include "mem/allocator.h" +#ifdef HAS_CUDA +#include +#include "helper_cuda.h" +#endif namespace jittor {