Allow datatypes besides fp32 in conv2d_transpose for cuda. (apache#6593)

jinqiua · Oct 1, 2020 · e31564e · e31564e
1 parent e78aa61
commit e31564e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/topi/cuda/conv2d_transpose_nchw.py b/python/tvm/topi/cuda/conv2d_transpose_nchw.py
@@ -96,7 +96,7 @@ def conv2d_transpose_nchw(cfg, data, kernel, stride, padding, out_dtype, output_
                 tvm.tir.indexdiv(y - pad_top, stride_height),
                 tvm.tir.indexdiv(x - pad_left, stride_width),
             ],
-            tvm.tir.const(0.0, "float32"),
+            tvm.tir.const(0.0, data.dtype),
         ),
         name="data_pad",
     )