From 0a8addf802e2198084b9cc66e49ca4ae2fdd8125 Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Wed, 13 Dec 2017 15:24:53 +0800
Subject: [PATCH] Make cast op support bool (#6562)

Also add `elemwise_sub/mul/abs/clip` layers
---
 paddle/operators/cast_op.cc      | 3 ++-
 paddle/operators/cast_op.cu      | 3 ++-
 python/paddle/v2/fluid/layers.py | 5 +++--
 3 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/paddle/operators/cast_op.cc b/paddle/operators/cast_op.cc
index 42bff69a1e..d641b8fc9f 100644
--- a/paddle/operators/cast_op.cc
+++ b/paddle/operators/cast_op.cc
@@ -74,4 +74,5 @@ REGISTER_OP_WITH_KERNEL(cast, ops::CastOpGradMaker, ops::CastOpInferShape,
 REGISTER_OP_CPU_KERNEL(cast, ops::CastOpKernel<CPU, float>,
                        ops::CastOpKernel<CPU, double>,
                        ops::CastOpKernel<CPU, int>,
-                       ops::CastOpKernel<CPU, int64_t>);
+                       ops::CastOpKernel<CPU, int64_t>,
+                       ops::CastOpKernel<CPU, bool>);
diff --git a/paddle/operators/cast_op.cu b/paddle/operators/cast_op.cu
index 4681deaa62..91e6fb391c 100644
--- a/paddle/operators/cast_op.cu
+++ b/paddle/operators/cast_op.cu
@@ -19,4 +19,5 @@ using CastOpKernel =
     paddle::operators::CastOpKernel<paddle::platform::CUDADeviceContext, T>;
 
 REGISTER_OP_CUDA_KERNEL(cast, CastOpKernel<float>, CastOpKernel<double>,
-                        CastOpKernel<int>, CastOpKernel<int64_t>);
+                        CastOpKernel<int>, CastOpKernel<int64_t>,
+                        CastOpKernel<bool>);
diff --git a/python/paddle/v2/fluid/layers.py b/python/paddle/v2/fluid/layers.py
index 9f5a219b20..f67d6d08c7 100644
--- a/python/paddle/v2/fluid/layers.py
+++ b/python/paddle/v2/fluid/layers.py
@@ -15,8 +15,9 @@ __all__ = [
 ]
 
 _REGISTER_LAYER_FROM_OPS = [
-    'mean', 'mul', 'elementwise_add', 'elementwise_div', 'dropout', 'reshape',
-    'sigmoid', 'scale', 'transpose', 'sigmoid_cross_entropy_with_logits'
+    'mean', 'mul', 'dropout', 'reshape', 'sigmoid', 'scale', 'transpose',
+    'sigmoid_cross_entropy_with_logits', 'elementwise_add', 'elementwise_div',
+    'elementwise_sub', 'elementwise_mul', 'clip', 'abs'
 ]
 
 for _OP in set(_REGISTER_LAYER_FROM_OPS):
-- 
GitLab