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

fix

This commit is contained in:
cxjyxx_me 2020-12-07 22:48:34 +08:00
parent 6f5a4cb336
commit 3221ac122d
3 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -232,7 +232,7 @@ void CudnnConvBackwardWOp::jit_run() {
best_algo_idx=i; best_algo_idx=i;
break; break;
} }
ASSERT(best_algo_idx!=-1);// TODO: assert error when depthconv ASSERT(best_algo_idx!=-1);
algo=perf_results[best_algo_idx].algo; algo=perf_results[best_algo_idx].algo;
if (benchmark) { if (benchmark) {
bwdw_algo_cache[jk.to_string()] = algo; bwdw_algo_cache[jk.to_string()] = algo;

9
python/jittor/depthwise_conv.py
View File

@ -1,3 +1,12 @@
# ***************************************************************
# Copyright (c) 2020 Jittor. Authors:
# Guoye Yang <498731903@qq.com>
# Dun Liang <randonlang@gmail.com>.
#
# All Rights Reserved.
# This file is subject to the terms and conditions defined in
# file 'LICENSE.txt', which is part of this source code package.
# ***************************************************************
import jittor as jt import jittor as jt
from jittor import init from jittor import init
from jittor import nn from jittor import nn

5
python/jittor/test/test_depthwise_conv.py
View File

@ -1,5 +1,8 @@
# *************************************************************** # ***************************************************************
# Copyright (c) 2020 Jittor. Authors: Dun Liang <randonlang@gmail.com>. All Rights Reserved. # Copyright (c) 2020 Jittor. Authors:
# Guoye Yang <498731903@qq.com>
# Dun Liang <randonlang@gmail.com>.
# All Rights Reserved.
# This file is subject to the terms and conditions defined in # This file is subject to the terms and conditions defined in
# file 'LICENSE.txt', which is part of this source code package. # file 'LICENSE.txt', which is part of this source code package.
# *************************************************************** # ***************************************************************