Merge pull request #514 from ngc92/compile-fix

fix compilation with older nvcc
PistonDevelopers · Jun 2, 2024 · cb5eff0 · cb5eff0
2 parents ff98d8b + a902323
commit cb5eff0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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) {