diff options
| author |  Yangzihao Wang <yangzihao@google.com> | 2017-05-28 14:20:19 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-05-28 14:23:23 -0700 |
| commit | e78e5ec8a8c862e65b6a194e9caea377120d7207 (patch) | |
| tree | ef713546bfdacced118056167b49de0c4c2f1a36 /tensorflow/stream_executor/stream_executor_pimpl.h | |
| parent | 3f9b69a50f40154f6078e1610ce7d3afa94bd07c (diff) | |
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
Diffstat (limited to 'tensorflow/stream_executor/stream_executor_pimpl.h')
| -rw-r--r-- | tensorflow/stream_executor/stream_executor_pimpl.h | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/stream_executor/stream_executor_pimpl.h b/tensorflow/stream_executor/stream_executor_pimpl.h index 5c52afa794..3dbeddd5d4 100644 --- a/tensorflow/stream_executor/stream_executor_pimpl.h +++ b/tensorflow/stream_executor/stream_executor_pimpl.h @@ -342,15 +342,18 @@ class StreamExecutor { bool SupportsDnn() const; // Get the list of supported algorithms for the forward convolution opeartion. - bool GetConvolveAlgorithms(std::vector<dnn::AlgorithmType> *out_algorithms); + bool GetConvolveAlgorithms(bool with_winograd_nonfused, + std::vector<dnn::AlgorithmType> *out_algorithms); // Get the list of supported algorithms for the backward convolution on data. bool GetConvolveBackwardDataAlgorithms( + bool with_winograd_nonfused, std::vector<dnn::AlgorithmType> *out_algorithms); // Get the list of supported algorithms for the backward convolution on the // filter. bool GetConvolveBackwardFilterAlgorithms( + bool with_winograd_nonfused, std::vector<dnn::AlgorithmType> *out_algorithms); // Get the list of supported algorithms for BLAS gemm. |