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

fix copy op

This commit is contained in:
Dun Liang 2020-07-23 10:38:22 +08:00
parent ad320ed2cf
commit 8bb698c225
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/jittor/__init__.py
View File

@ -7,7 +7,7 @@
# This file is subject to the terms and conditions defined in
# file 'LICENSE.txt', which is part of this source code package.
# ***************************************************************
__version__ = '1.1.6.3'
__version__ = '1.1.6.4'
from . import lock
with lock.lock_scope():
from . import compiler

6
src/ops/copy_op.cc
View File

@ -8,6 +8,10 @@
#include "var.h"
#include "ops/op_register.h"
#include "ops/copy_op.h"
#ifdef HAS_CUDA
#include <cuda_runtime.h>
#include <helper_cuda.h>
#endif
namespace jittor {
@ -37,7 +41,7 @@ void CopyOp::run() {
}
#ifdef HAS_CUDA
else {
std::cudaMemcpyAsync(y_ptr, x_ptr, size, cudaMemcpyDefault, 0);
checkCudaErrors(cudaMemcpyAsync(y_ptr, x_ptr, size, cudaMemcpyDefault, 0));
}
#endif
}