[ROCM] Fix conv2d (#3107)

d8f547c7 · Bing Xu · Tianqi Chen · 8f56949b · d8f547c7
Commit d8f547c7 authored Apr 26, 2019 by Bing Xu Committed by Tianqi Chen Apr 26, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 additions and 1 deletions

topi/python/topi/rocm/conv2d.py
+1 -1

No files found.
--- a/topi/python/topi/rocm/conv2d.py
+++ b/topi/python/topi/rocm/conv2d.py
@@ -71,7 +71,7 @@ def conv2d_rocm(cfg, data, kernel, strides, padding, dilation, layout='NCHW', ou
                    ((KW - 1) * dilation_w + 1))

        return miopen.conv2d_forward(data,
-                                     kernel_before_dilation,
+                                     kernel,
                                     stride_h,
                                     stride_w,
                                     pad_h,