From 1f5cd1a7fd9563e8d2efced5a224b8b6b02abb6d Mon Sep 17 00:00:00 2001
From: Omar Elayan <oelayan@habana.ai>
Date: Tue, 10 Sep 2024 11:46:41 +0300
Subject: [PATCH 1/3] wrap include cuda_bf16.h with ifdef BF16_AVAILABLE

---
 csrc/fp_quantizer/fp_quantize.cu         | 2 ++
 csrc/fp_quantizer/includes/fp_quantize.h | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/csrc/fp_quantizer/fp_quantize.cu b/csrc/fp_quantizer/fp_quantize.cu
index cca063956167..66ea7392e011 100644
--- a/csrc/fp_quantizer/fp_quantize.cu
+++ b/csrc/fp_quantizer/fp_quantize.cu
@@ -15,7 +15,9 @@
 #include <cuda_fp16.h>
 #include <curand_kernel.h>
 
+#ifdef BF16_AVAILABLE
 #include <cuda_bf16.h>
+#endif
 #include <cuda_runtime_api.h>
 
 using ROp = reduce::ROpType;
diff --git a/csrc/fp_quantizer/includes/fp_quantize.h b/csrc/fp_quantizer/includes/fp_quantize.h
index 507252d6e722..75ffc435aec8 100644
--- a/csrc/fp_quantizer/includes/fp_quantize.h
+++ b/csrc/fp_quantizer/includes/fp_quantize.h
@@ -10,7 +10,9 @@
 
 #include <cuda_fp16.h>
 
+#ifdef BF16_AVAILABLE
 #include <cuda_bf16.h>
+#endif#include <cuda_bf16.h>
 #include <cuda_runtime_api.h>
 #include <stdio.h>
 

From 9c0d5958f370267473f83b5ea92ee0b435dd3f7f Mon Sep 17 00:00:00 2001
From: Omar Elayan <oelayan@habana.ai>
Date: Tue, 10 Sep 2024 12:09:20 +0300
Subject: [PATCH 2/3] formatting issue

---
 csrc/fp_quantizer/includes/fp_quantize.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/csrc/fp_quantizer/includes/fp_quantize.h b/csrc/fp_quantizer/includes/fp_quantize.h
index 75ffc435aec8..94dadcff8ed4 100644
--- a/csrc/fp_quantizer/includes/fp_quantize.h
+++ b/csrc/fp_quantizer/includes/fp_quantize.h
@@ -12,7 +12,7 @@
 
 #ifdef BF16_AVAILABLE
 #include <cuda_bf16.h>
-#endif#include <cuda_bf16.h>
+#endif #include < cuda_bf16.h>
 #include <cuda_runtime_api.h>
 #include <stdio.h>
 

From a61c2983d5d28202f3025c113005e77a116a6e95 Mon Sep 17 00:00:00 2001
From: Omar Elayan <oelayan@habana.ai>
Date: Tue, 10 Sep 2024 12:11:55 +0300
Subject: [PATCH 3/3] formatting issue

---
 csrc/fp_quantizer/includes/fp_quantize.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/csrc/fp_quantizer/includes/fp_quantize.h b/csrc/fp_quantizer/includes/fp_quantize.h
index 94dadcff8ed4..60c75541f603 100644
--- a/csrc/fp_quantizer/includes/fp_quantize.h
+++ b/csrc/fp_quantizer/includes/fp_quantize.h
@@ -12,7 +12,7 @@
 
 #ifdef BF16_AVAILABLE
 #include <cuda_bf16.h>
-#endif #include < cuda_bf16.h>
+#endif
 #include <cuda_runtime_api.h>
 #include <stdio.h>