aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/conv_ops_gpu.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/kernels/conv_ops_gpu.h')
-rw-r--r--tensorflow/core/kernels/conv_ops_gpu.h5
1 files changed, 2 insertions, 3 deletions
diff --git a/tensorflow/core/kernels/conv_ops_gpu.h b/tensorflow/core/kernels/conv_ops_gpu.h
index 4215c4541c..d2c8020bb6 100644
--- a/tensorflow/core/kernels/conv_ops_gpu.h
+++ b/tensorflow/core/kernels/conv_ops_gpu.h
@@ -139,9 +139,8 @@ class ConvParameters {
bool ShouldIncludeWinogradNonfusedAlgo(
se::StreamExecutor* stream_exec) const {
// Skip this check for cuDNN 7 and newer.
- se::port::StatusOr<std::tuple<int, int, int>> version =
- stream_exec->AsDnn()->GetVersion();
- if (version.ok() && std::get<0>(version.ValueOrDie()) >= 7) {
+ auto version = stream_exec->AsDnn()->GetVersion();
+ if (version.ok() && version.ValueOrDie().major_version() >= 7) {
return true;
}
return ShouldIncludeWinogradNonfusedAlgoPreCudnn7<T>();