nit

NVIDIA · Jul 19, 2023 · 92a18e9 · 92a18e9
1 parent dafee80
commit 92a18e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/cuquantum_benchmarks/_utils.py b/benchmarks/cuquantum_benchmarks/_utils.py
@@ -456,7 +456,7 @@ class L2flush:
     https://github.com/NVIDIA/nvbench/blob/main/nvbench/detail/l2flush.cuh.
     """
     def __init__(self):
-        self.l2_size = 3 * cp.cuda.Device().attributes['L2CacheSize']
+        self.l2_size = cp.cuda.Device().attributes['L2CacheSize']
         self.mem = cp.cuda.alloc(self.l2_size) if self.l2_size > 0 else None
 
     def flush(self):