lower plevel of conv2d winograd on cuda (apache#4987)

neo-ai · Apr 17, 2020 · ef53274 · ef53274
1 parent ec27f71
commit ef53274
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/op/strategy/cuda.py b/python/tvm/relay/op/strategy/cuda.py
@@ -98,7 +98,7 @@ def conv2d_strategy_cuda(attrs, inputs, out_type, target):
                     wrap_compute_conv2d(topi.cuda.conv2d_nchw_winograd),
                     wrap_topi_schedule(topi.cuda.schedule_conv2d_nchw_winograd),
                     name="conv2d_nchw_winograd.cuda",
-                    plevel=15)
+                    plevel=5)
         elif layout == "HWCN":
             assert kernel_layout == "HWIO"
             strategy.add_implementation(