add peek_logged

2020-11-30 16:42:46 +08:00 · 2020-11-30 16:42:46 +08:00 · 3b4430906d
parent 9f3c97a605
commit 3b4430906d
2 changed files with 4 additions and 1 deletions
--- a/extern/cuda/src/helper_cuda.cc
+++ b/extern/cuda/src/helper_cuda.cc
@ -15,7 +15,6 @@
 #include <cuda_runtime.h>
 #include "helper_cuda.h"

-
 #ifdef _CUFFT_H_
 // cuFFT API errors
 const char *_cudaGetErrorEnum(cufftResult error) {
--- 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 <cuda_runtime.h>
+#include "helper_cuda.h"
+#endif

 namespace jittor {