This is an automated email from the ASF dual-hosted git repository.
masahi pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git
The following commit(s) were added to refs/heads/master by this push:
new 7eed17b lower plevel of conv2d winograd on cuda (#4987)
7eed17b is described below
commit 7eed17b92e71ad199e9ea5bc094221b20773b4d4
Author: Haichen Shen <[email protected]>
AuthorDate: Sun Mar 8 04:51:02 2020 -0700
lower plevel of conv2d winograd on cuda (#4987)
---
python/tvm/relay/op/strategy/cuda.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/python/tvm/relay/op/strategy/cuda.py
b/python/tvm/relay/op/strategy/cuda.py
index b2f559f..b1e77bd 100644
--- 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(
