From 68d2ce06010fd4e1548fc109308be1d8256c25c7 Mon Sep 17 00:00:00 2001
From: lancer <tangshao28@gmail.com>
Date: Wed, 10 Apr 2024 13:35:20 -0700
Subject: [PATCH] fix error in small typos in matmul_forward.cu

---
 dev/cuda/matmul_forward.cu | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dev/cuda/matmul_forward.cu b/dev/cuda/matmul_forward.cu
index 81d53bf..4dbf582 100644
--- a/dev/cuda/matmul_forward.cu
+++ b/dev/cuda/matmul_forward.cu
@@ -239,12 +239,12 @@ int main(int argc, char **argv) {
             printf("%f %f\n", out[i], out_gpu[i]);
         }
         // ensure correctness for all elements
-        if (i >= 5 && fabs(out[i] - out_gpu[i]) > 1e-4) {
+        if (fabs(out[i] - out_gpu[i]) > 1e-4) {
             printf("Mismatch at %d: %f vs %f\n", i, out[i], out_gpu[i]);
             exit(1);
         }
     }
-    printf("Results match at block_size=256!\n");
+    printf("Results match at block_size=1024!\n");
 
     // time the kernel at different block sizes
     int sqrt_block_sizes[] = {4, 8, 16, 32};