fix testcase

Signed-off-by: Cruz Zhao <[email protected]>

Author: zxpdemonio

scripts/test_tensor_api.py

@@ -511,8 +511,8 @@ def test_07_batch_put_get_into_with_tp(self):
 class TestMooncakeBenchmark(MooncakeTestBase):
     # Benchmark Settings
     BENCH_ITERATIONS = 5
-    TENSOR_SIZE_MB = 64
-    TOTAL_SIZE_GB = 1 
+    TENSOR_SIZE_MB = 16
+    TOTAL_SIZE_GB = 256
 
     def setUp(self):
         """Benchmark-specific setUp."""
@@ -585,7 +585,7 @@ def test_benchmark_02_tp_batch(self):
 
     def test_benchmark_03_batch_put_get_into(self):
         """Benchmark: Zero copy Batch Get."""
-        buffer_spacing = 1024 * 1024 * 1024  # 1GB per tensor slot
+        buffer_spacing = 512 * 1024 * 1024  # 1GB per tensor slot
         batch_size = len(self.keys)
         total_buffer_size = buffer_spacing * batch_size
 
@@ -645,7 +645,7 @@ def test_benchmark_04_batch_put_get_into_with_tp(self):
         tp_size = 4
         split_dim = 0
         batch_size = len(self.keys)
-        buffer_spacing = 1024 * 1024 * 1024  # 1GB per tensor slot
+        buffer_spacing = 512 * 1024 * 1024  # 1GB per tensor slot
 
         # Allocate and register a separate buffer for each TP rank
         rank_buffers = []  # Store metadata for cleanup