From 72f9cda0191d9b0473580c54cf54aca428b0d285 Mon Sep 17 00:00:00 2001
From: Abdelrauf <qwr@live.ru>
Date: Tue, 18 Feb 2020 18:01:43 +0400
Subject: [PATCH] Added missing bfloat16 (#252)

Signed-off-by: AbdelRauf <rauf@konduit.ai>
---
 libnd4j/include/platformmath.h | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/libnd4j/include/platformmath.h b/libnd4j/include/platformmath.h
index b7cbe3745..b58e8f7f6 100644
--- a/libnd4j/include/platformmath.h
+++ b/libnd4j/include/platformmath.h
@@ -326,6 +326,11 @@ namespace nd4j {
 #endif
         }
 
+        template <>
+        math_def FORCEINLINE bfloat16 p_floor(bfloat16 value) {
+            return static_cast<bfloat16>(floorf((float)value));
+        }
+
         template <>
         math_def FORCEINLINE double p_floor(double value) {
             return floor(value);
@@ -352,6 +357,11 @@ namespace nd4j {
 #endif
         }
 
+        template <>
+        math_def FORCEINLINE bfloat16 p_ceil(bfloat16 value) {
+            return static_cast<bfloat16>(ceilf((float)value));
+        }
+
         template <>
         math_def FORCEINLINE double p_ceil(double value) {
             return ceil(value);
@@ -374,6 +384,12 @@ namespace nd4j {
             return static_cast<float16>(roundf((float) val));
         }
 
+        template <>
+        math_def FORCEINLINE bfloat16 p_round(bfloat16 value) {
+            return static_cast<bfloat16>(roundf((float)value));
+        }
+
+
         template <>
         math_def FORCEINLINE double p_round(double value) {
             return round(value);