diff options
author | Yangzihao Wang <yangzihao@google.com> | 2017-12-20 17:48:42 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-20 17:52:33 -0800 |
commit | 443b4a3b115f63523617871f55460791de202e30 (patch) | |
tree | 5e93fb48393b30bfa315c52ef3dee91881dc52fd /tensorflow/contrib/fused_conv | |
parent | ba554ed697e4c7364f3cf5742df873ce9686333f (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.cc | 2 |
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.")); |