提交 813bc1e9 authored 作者: --global's avatar --global

Add config flag for algo selection for gradient convolution

上级 65eb53a5
......@@ -350,7 +350,13 @@ class GpuDnnConvDesc(GpuOp):
AddConfigVar('dnn.conv.workmem',
"Default value for the workmem attribute of cudnn convolutions.",
EnumStr('small', 'none', 'large', 'guess', 'time'),
EnumStr('small', 'none', 'large', 'fft', 'guess', 'time'),
in_c_key=False)
AddConfigVar('dnn.conv.workmem_bwd',
"Default value for the workmem attribute of cudnn gradient "
"convolutions.",
EnumStr('none', 'deterministic', 'fft', 'guess'),
in_c_key=False)
# scalar constants
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论