aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/fused_conv
diff options
context:
space:
mode:
authorGravatar Yangzihao Wang <yangzihao@google.com>2017-12-20 17:48:42 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-12-20 17:52:33 -0800
commit443b4a3b115f63523617871f55460791de202e30 (patch)
tree5e93fb48393b30bfa315c52ef3dee91881dc52fd /tensorflow/contrib/fused_conv
parentba554ed697e4c7364f3cf5742df873ce9686333f (diff)
Fix the condition check for int8 fused_conv so that it works on Volta.
PiperOrigin-RevId: 179756490
Diffstat (limited to 'tensorflow/contrib/fused_conv')
-rw-r--r--tensorflow/contrib/fused_conv/kernels/fused_conv2d_bias_activation_op.cc2
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/fused_conv/kernels/fused_conv2d_bias_activation_op.cc b/tensorflow/contrib/fused_conv/kernels/fused_conv2d_bias_activation_op.cc
index 5fec69ea43..dc9e5558ed 100644
--- a/tensorflow/contrib/fused_conv/kernels/fused_conv2d_bias_activation_op.cc
+++ b/tensorflow/contrib/fused_conv/kernels/fused_conv2d_bias_activation_op.cc
@@ -303,7 +303,7 @@ void LaunchFusedConv2DBiasActivationOp<GPUDevice, T, BiasType, ScaleType>::
stream->parent()->GetDeviceDescription().cuda_compute_capability(&cc_major,
&cc_minor);
OP_REQUIRES(
- ctx, cc_major >= 6 && cc_minor >= 1,
+ ctx, ((cc_major == 6 && cc_minor >= 1) || cc_major > 6),
errors::Unimplemented(
"FusedConv2DBiasActivation for int8 is only supported on GPUs with "
"compute capability 6.1 or later."));