[incubator-mxnet] branch master updated: fix CreateOp to accept dtypes of uint8 and int32 (#9561)

[jx...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

jxie pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/master by this push:
     new 9aaa29d  fix CreateOp to accept dtypes of uint8 and int32 (#9561)
9aaa29d is described below

commit 9aaa29d1ce6f4db2f4daf47eabbb44dd4a75fcdb
Author: Soonhwan-Kwon <So...@users.noreply.github.com>
AuthorDate: Fri Jan 26 03:55:08 2018 +0900

    fix CreateOp to accept dtypes of uint8 and int32 (#9561)
---
 src/operator/slice_channel.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/operator/slice_channel.cu b/src/operator/slice_channel.cu
index adc8741..bc5ca52 100644
--- a/src/operator/slice_channel.cu
+++ b/src/operator/slice_channel.cu
@@ -31,7 +31,7 @@ namespace op {
 template<>
 Operator* CreateOp<gpu>(SliceChannelParam param, int dtype) {
   Operator* op = nullptr;
-  MSHADOW_REAL_TYPE_SWITCH(dtype, DType, {
+  MSHADOW_TYPE_SWITCH(dtype, DType, {
     op = new SliceChannelOp<gpu, DType>(param);
   })
   return op;

-- 
To stop receiving notification emails like this one, please contact
jxie@apache.org.