From 2c8de2677154044969af20ded84f94f3e67effd7 Mon Sep 17 00:00:00 2001
From: Muhammad Awad <112003944+mawad-amd@users.noreply.github.com>
Date: Thu, 28 Aug 2025 20:27:41 -0700
Subject: [PATCH] Update buffer size comment in README.md

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index bc696893..ead335e3 100644
--- a/README.md
+++ b/README.md
@@ -59,11 +59,11 @@ def kernel(buffer, buffer_size: tl.constexpr, block_size: tl.constexpr, heap_bas
 
 # Iris initialization
 heap_size = 2**30   # 1GiB symmetric heap for inter-GPU communication
-buffer_size = 4096  # 4KB buffer for demonstration
 iris_ctx = iris.iris(heap_size)
 cur_rank = iris_ctx.get_rank()
 
 # Iris tensor allocation
+buffer_size = 4096  # 4K elements buffer
 buffer = iris_ctx.zeros(buffer_size, device="cuda", dtype=torch.float32)
 
 # Launch the kernel on rank 0