From 6b7b5aae6fd0e1cb15ba25df6c5ba0c03671f28b Mon Sep 17 00:00:00 2001
From: Tunji Ruwase <olruwase@microsoft.com>
Date: Fri, 10 Feb 2023 13:09:52 +0000
Subject: [PATCH] Reduce I/O size

---
 tests/unit/ops/aio/test_aio.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/unit/ops/aio/test_aio.py b/tests/unit/ops/aio/test_aio.py
index 02417021e512..20223b53baf9 100644
--- a/tests/unit/ops/aio/test_aio.py
+++ b/tests/unit/ops/aio/test_aio.py
@@ -7,10 +7,10 @@
 from deepspeed.ops.aio import AsyncIOBuilder
 from unit.common import DistributedTest
 
-MEGA_BYTE = 1024**2
-BLOCK_SIZE = MEGA_BYTE
+KILO_BYTE = 1024
+BLOCK_SIZE = KILO_BYTE
 QUEUE_DEPTH = 2
-IO_SIZE = 4 * 1024
+IO_SIZE = 4 * BLOCK_SIZE
 IO_PARALLEL = 2
 
 if not deepspeed.ops.__compatible_ops__[AsyncIOBuilder.NAME]: