提交 9a9c5fb8 authored 作者: Frederic Bastien's avatar Frederic Bastien 提交者: Mathieu Germain

GpuDnnConv3d support small algo in v4.

上级 0d640a74
...@@ -497,10 +497,7 @@ class GpuDnnConv3d(GpuDnnConv): ...@@ -497,10 +497,7 @@ class GpuDnnConv3d(GpuDnnConv):
good_algo = ['none', 'small', 'fft_tiling', good_algo = ['none', 'small', 'fft_tiling',
'guess_once', 'guess_on_shape_change', 'guess_once', 'guess_on_shape_change',
'time_once', 'time_on_shape_change'] 'time_once', 'time_on_shape_change']
if version() < (5000, 5000): if algo is None and config.dnn.conv.algo_fwd not in good_algo:
# Need to confirm when small was added for conv3d.
algo = 'none'
elif algo is None and config.dnn.conv.algo_fwd not in good_algo:
algo = 'guess_once' algo = 'guess_once'
elif algo is not None and algo not in good_algo: elif algo is not None and algo not in good_algo:
algo = 'guess_once' algo = 'guess_once'
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论