diff --git a/python/tvm/autotvm/tophub.py b/python/tvm/autotvm/tophub.py
index 9ec9becc7245..43d7c7050c07 100644
--- a/python/tvm/autotvm/tophub.py
+++ b/python/tvm/autotvm/tophub.py
@@ -28,7 +28,7 @@
     'opencl':  "v0.02",
     'mali':    "v0.04",
 
-    'vta':     "v0.04",
+    'vta':     "v0.05",
 }
 
 logger = logging.getLogger('autotvm')
diff --git a/vta/python/vta/top/op.py b/vta/python/vta/top/op.py
index 34897ac83780..1e654f2300ee 100644
--- a/vta/python/vta/top/op.py
+++ b/vta/python/vta/top/op.py
@@ -72,7 +72,7 @@ def compute_conv2d(attrs, inputs, out):
                 kshape[-1] *= w_pack_factor
                 inputs[1] = reinterpret(inputs[1], kshape, dtype=env.wgt_dtype)
 
-            return topi.nn.conv2d(inputs[0], inputs[1], strides, padding, layout, out_dtype)
+            return topi.nn.conv2d(inputs[0], inputs[1], strides, padding, dilation, layout, out_dtype)
         else:
             return packed_group_conv2d(inputs[0], inputs[1], padding, strides, groups, out_dtype)
 
diff --git a/vta/python/vta/top/vta_conv2d.py b/vta/python/vta/top/vta_conv2d.py
index 4f1ab3fc12ca..96ef596048a2 100644
--- a/vta/python/vta/top/vta_conv2d.py
+++ b/vta/python/vta/top/vta_conv2d.py
@@ -11,10 +11,11 @@
 
 
 @autotvm.register_topi_compute(topi.nn.conv2d, 'vta', 'direct')
-def packed_conv2d(cfg, data, kernel, strides, padding, layout, out_dtype):
+def packed_conv2d(cfg, data, kernel, strides, padding, dilation, layout, out_dtype):
     """ Packed conv2d function."""
     if not is_packed_layout(layout):
         raise topi.InvalidShapeError()
+    assert dilation == (1, 1)
 
     if padding[0]:
         pad_data = topi.nn.pad(data, [0, 0, padding[0], padding[1], 0, 0], name="pad_data")