From e78e5ec8a8c862e65b6a194e9caea377120d7207 Mon Sep 17 00:00:00 2001 From: Yangzihao Wang Date: Sun, 28 May 2017 14:20:19 -0700 Subject: Set winograd nofused flag to be true by default. Disable winograd nonfused conv for certain input params to avoid a known bug in cuDNNv5 and cuDNNv6. PiperOrigin-RevId: 157352847 --- tensorflow/stream_executor/dnn.h | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'tensorflow/stream_executor/dnn.h') diff --git a/tensorflow/stream_executor/dnn.h b/tensorflow/stream_executor/dnn.h index c5805064f3..8e56933ba3 100644 --- a/tensorflow/stream_executor/dnn.h +++ b/tensorflow/stream_executor/dnn.h @@ -952,7 +952,7 @@ class DnnSupport { // Return a list of algorithms supported by the forward convolution pass. virtual bool GetConvolveAlgorithms( - std::vector* out_algorithms); + bool with_winograd_nonfused, std::vector* out_algorithms); // Enqueues a double-precision convolution operation onto the stream. // See DoConvolve above for argument details. @@ -1056,7 +1056,7 @@ class DnnSupport { // Return a list of algorithms supported by the backward convolution pass for // data. virtual bool GetConvolveBackwardDataAlgorithms( - std::vector* out_algorithms); + bool with_winograd_nonfused, std::vector* out_algorithms); virtual bool DoConvolveBackwardData( Stream* stream, const FilterDescriptor& filter_descriptor, @@ -1104,7 +1104,7 @@ class DnnSupport { // Return a list of algorithms supported by the backward convolution pass for // filters. virtual bool GetConvolveBackwardFilterAlgorithms( - std::vector* out_algorithms); + bool with_winograd_nonfused, std::vector* out_algorithms); virtual bool DoConvolveBackwardFilter( Stream* stream, const BatchDescriptor& input_descriptor, -- cgit v1.2.3