From 915a6d6e44b838e7618e56021bf8dee6163b6ff0 Mon Sep 17 00:00:00 2001
From: Chip Kerchner <chip.kerchner@ibm.com>
Date: Thu, 3 Oct 2024 14:08:21 -0500
Subject: [PATCH] Add casting.

---
 kernel/power/sbgemv_common.c | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/kernel/power/sbgemv_common.c b/kernel/power/sbgemv_common.c
index 47de837cc..8ad7f92e7 100644
--- a/kernel/power/sbgemv_common.c
+++ b/kernel/power/sbgemv_common.c
@@ -121,12 +121,12 @@ FORCEINLINE void copy_x(BLASLONG n, IFLOAT *src, IFLOAT *dest, BLASLONG inc_src)
 
 FORCEINLINE void copy_y_beta(BLASLONG n, FLOAT *src, FLOAT *dest, BLASLONG inc_src, FLOAT beta)
 {
-  if (beta == 0) {
+  if (beta == (FLOAT)0) {
     for (BLASLONG i = 0; i < n; i++) {
       *dest++ = (FLOAT)0;
       src += inc_src;
     }
-  } else if (beta == 1) {
+  } else if (beta == (FLOAT)1) {
     for (BLASLONG i = 0; i < n; i++) {
       *dest++ = *src;
       src += inc_src;
@@ -141,12 +141,12 @@ FORCEINLINE void copy_y_beta(BLASLONG n, FLOAT *src, FLOAT *dest, BLASLONG inc_s
 
 FORCEINLINE void copy_y(BLASLONG n, FLOAT *src, FLOAT *dest, BLASLONG inc_src, FLOAT beta)
 {
-  if (beta == 0) {
+  if (beta == (FLOAT)0) {
     for (BLASLONG i = 0; i < n; i++) {
       *dest = *src++;
       dest += inc_src;
     }
-  } else if (beta == 1) {
+  } else if (beta == (FLOAT)1) {
     for (BLASLONG i = 0; i < n; i++) {
       *dest += *src++;
       dest += inc_src;
@@ -169,9 +169,9 @@ FORCEINLINE void move_y(BLASLONG n, FLOAT *src, FLOAT *dest, BLASLONG inc_dest)
 
 static void BF16GEMV_N_beta(BLASLONG n, FLOAT *output_vector, FLOAT *input_vector, FLOAT beta)
 {
-  if (beta == 0) {
+  if (beta == (FLOAT)0) {
     memset(output_vector, 0, sizeof(FLOAT) * n);
-  } else if (beta == 1) {
+  } else if (beta == (FLOAT)1) {
     if (output_vector != input_vector) {
       memcpy(output_vector, input_vector, sizeof(FLOAT) * n);
     }