This is an automated email from the ASF dual-hosted git repository.
tqchen 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 e31564e Allow datatypes besides fp32 in conv2d_transpose for cuda.
(#6593)
e31564e is described below
commit e31564e1e3b729d70bd5b0866a84497b05cae6ae
Author: Josh Fromm <[email protected]>
AuthorDate: Thu Oct 1 07:40:39 2020 -0700
Allow datatypes besides fp32 in conv2d_transpose for cuda. (#6593)
---
python/tvm/topi/cuda/conv2d_transpose_nchw.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/python/tvm/topi/cuda/conv2d_transpose_nchw.py
b/python/tvm/topi/cuda/conv2d_transpose_nchw.py
index 46ee685..915e6cd 100644
--- 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",
)
