lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add peek_logged

This commit is contained in:
li-xl 2020-11-30 16:42:46 +08:00 committed by Dun Liang
parent 9f3c97a605
commit 3b4430906d
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
extern/cuda/src/helper_cuda.cc vendored
View File

@ -15,7 +15,6 @@
#include <cuda_runtime.h> #include <cuda_runtime.h>
#include "helper_cuda.h" #include "helper_cuda.h"
#ifdef _CUFFT_H_ #ifdef _CUFFT_H_
// cuFFT API errors // cuFFT API errors
const char *_cudaGetErrorEnum(cufftResult error) { const char *_cudaGetErrorEnum(cufftResult error) {

4
src/executor.h
View File

@ -7,6 +7,10 @@
#pragma once #pragma once
#include "common.h" #include "common.h"
#include "mem/allocator.h" #include "mem/allocator.h"
#ifdef HAS_CUDA
#include <cuda_runtime.h>
#include "helper_cuda.h"
#endif
namespace jittor { namespace jittor {