Skip to content

T
tic
Unverified Commit 7eed17b9 by Haichen Shen Committed by GitHub
Options

lower plevel of conv2d winograd on cuda (#4987)

parent 87faaf12
Showing with 1 additions and 1 deletions
python/tvm/relay/op/strategy/cuda.py
...@@ -98,7 +98,7 @@ def conv2d_strategy_cuda(attrs, inputs, out_type, target): ...@@ -98,7 +98,7 @@ def conv2d_strategy_cuda(attrs, inputs, out_type, target):
wrap_compute_conv2d(topi.cuda.conv2d_nchw_winograd), wrap_compute_conv2d(topi.cuda.conv2d_nchw_winograd),
wrap_topi_schedule(topi.cuda.schedule_conv2d_nchw_winograd), wrap_topi_schedule(topi.cuda.schedule_conv2d_nchw_winograd),
name="conv2d_nchw_winograd.cuda", name="conv2d_nchw_winograd.cuda",
plevel=15) plevel=5)
elif layout == "HWCN": elif layout == "HWCN":
assert kernel_layout == "HWIO" assert kernel_layout == "HWIO"
strategy.add_implementation( strategy.add_implementation(
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment