diff --git a/extern/cuda/cudnn/ops/cudnn_conv_backward_w_op.cc b/extern/cuda/cudnn/ops/cudnn_conv_backward_w_op.cc index 571f3ec3..d1b76e65 100644 --- a/extern/cuda/cudnn/ops/cudnn_conv_backward_w_op.cc +++ b/extern/cuda/cudnn/ops/cudnn_conv_backward_w_op.cc @@ -232,7 +232,7 @@ void CudnnConvBackwardWOp::jit_run() { best_algo_idx=i; break; } - ASSERT(best_algo_idx!=-1);// TODO: assert error when depthconv + ASSERT(best_algo_idx!=-1); algo=perf_results[best_algo_idx].algo; if (benchmark) { bwdw_algo_cache[jk.to_string()] = algo; diff --git a/python/jittor/depthwise_conv.py b/python/jittor/depthwise_conv.py index 72eb8f5d..6571837c 100644 --- a/python/jittor/depthwise_conv.py +++ b/python/jittor/depthwise_conv.py @@ -1,3 +1,12 @@ +# *************************************************************** +# Copyright (c) 2020 Jittor. Authors: +# Guoye Yang <498731903@qq.com> +# Dun Liang . +# +# 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 from jittor import init from jittor import nn diff --git a/python/jittor/test/test_depthwise_conv.py b/python/jittor/test/test_depthwise_conv.py index c6af5f1e..04465a9f 100644 --- a/python/jittor/test/test_depthwise_conv.py +++ b/python/jittor/test/test_depthwise_conv.py @@ -1,5 +1,8 @@ # *************************************************************** -# Copyright (c) 2020 Jittor. Authors: Dun Liang . All Rights Reserved. +# Copyright (c) 2020 Jittor. Authors: +# Guoye Yang <498731903@qq.com> +# Dun Liang . +# 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. # ***************************************************************