From a1e9c5a1454c35598080040da743a11b1cdc76c7 Mon Sep 17 00:00:00 2001 From: Tim Shen Date: Wed, 15 Jan 2020 18:26:05 -0800 Subject: [PATCH] Blacklist algorithms described in nvbugs/2774617. PiperOrigin-RevId: 289979133 Change-Id: Iefcf78590b3abc56143068e597e46888d1683f44 --- .../xla/service/gpu/hlo_algorithm_blacklist.cc | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/tensorflow/compiler/xla/service/gpu/hlo_algorithm_blacklist.cc b/tensorflow/compiler/xla/service/gpu/hlo_algorithm_blacklist.cc index bb85c509d18..38914ab9e0f 100644 --- a/tensorflow/compiler/xla/service/gpu/hlo_algorithm_blacklist.cc +++ b/tensorflow/compiler/xla/service/gpu/hlo_algorithm_blacklist.cc @@ -26,6 +26,20 @@ namespace gpu { // MSVC requires the extra const. Without, it reports an // "error C2131: expression did not evaluate to a constant". constexpr const absl::string_view kDefaultBlacklist = R"pb( + entries { + hlo: "(f32[4,32,32,32]{2,1,3,0}, u8[0]{0}) custom-call(f32[4,32,32,32]{2,1,3,0}, f32[5,5,32,32]{1,0,2,3}), window={size=5x5 pad=2_2x2_2}, dim_labels=b01f_01io->b01f, custom_call_target=\"__cudnn$convForward\", backend_config=\"{conv_result_scale:1}\"" + cc { major: 7 } + cudnn_version { major: 7 minor: 6 patch: 4 } + algos { id: 7 } + blas_version: "10201" + } + entries { + hlo: "(f32[4,32,32,32]{2,1,3,0}, u8[0]{0}) custom-call(f32[4,32,32,32]{2,1,3,0}, f32[5,5,32,32]{1,0,2,3}), window={size=5x5 pad=2_2x2_2}, dim_labels=b01f_01io->b01f, custom_call_target=\"__cudnn$convForward\", backend_config=\"{conv_result_scale:1}\"" + cc { major: 7 } + cudnn_version { major: 7 minor: 6 patch: 4 } + algos { id: 7 tensor_ops: true } + blas_version: "10201" + } )pb"; absl::Span