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

fix cudnn_backward_x_op and remove item

This commit is contained in:
li-xl 2020-11-26 17:36:59 +08:00
parent 342214d7f7
commit 2ca8b91122
2 changed files with 1 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
extern/cuda/cudnn/ops/cudnn_conv_backward_x_op.cc vendored
View File

@ -178,7 +178,7 @@ void CudnnConvBackwardXOp::jit_run() {
CUDNN_CONVOLUTION_BWD_DATA_ALGO_WINOGRAD,
CUDNN_CONVOLUTION_BWD_DATA_ALGO_WINOGRAD_NONFUSED
};
int num_algos = CUDNN_CONVOLUTION_BWD_FILTER_ALGO_COUNT;
int num_algos = CUDNN_CONVOLUTION_BWD_DATA_ALGO_COUNT;
int perf_count;
cudnnConvolutionBwdDataAlgoPerf_t perf_results[num_algos];
cudnnConvolutionBwdDataAlgo_t algo;

6
python/jittor/misc.py
View File

@ -443,12 +443,6 @@ def log2(x):
jt.Var.log2 = log2
def item(x):
assert x.ndim==1 and x.shape[0]==1
return x.numpy().item()
jt.Var.item = item
def meshgrid(*tensors):
r'''
Take N tensors, each of which can be 1-dimensional vector, and create N n-dimensional grids,