From a902323f365c0254f3166adeaeeb52985c718cc5 Mon Sep 17 00:00:00 2001
From: Erik Schultheis <erik.schultheis@aalto.fi>
Date: Sun, 2 Jun 2024 12:40:28 +0300
Subject: [PATCH] fix compilation with older nvcc

---
 train_gpt2.cu | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train_gpt2.cu b/train_gpt2.cu
index bf6acf1..a57d801 100644
--- a/train_gpt2.cu
+++ b/train_gpt2.cu
@@ -228,10 +228,10 @@ struct alignas(16) Packed128 {
         return result;
     }
     __device__ static Packed128 zeros() {
-        return constant(0);
+        return constant(0.f);
     }
     __device__ static Packed128 ones() {
-        return constant(1);
+        return constant(1.f);
     }
 
     __device__ ElementType& operator[](int index) {