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

polish

This commit is contained in:
Dun Liang 2020-03-29 23:10:30 +08:00
parent 16c7d3e58c
commit f2dd05c905
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/misc/cuda_flags.h
View File

@ -11,8 +11,7 @@ namespace jittor {
#ifdef HAS_CUDA
DECLARE_FLAG(int, use_cuda);
// #if CUDA_VERSION < 10000
#if 1
#if CUDA_VERSION < 10000
#define _cudaLaunchHostFunc(a,b,c) \
cudaStreamAddCallback(a,b,c,0)
#define CUDA_HOST_FUNC_ARGS cudaStream_t stream, cudaError_t status, void*