fix:解决测试config出现的num_nvl_bytes不足的bug

8f9cb191 · lishen · 8c44b1fa · 8f9cb191
Commit 8f9cb191 authored May 12, 2026 by lishen
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

deep_ep/buffer.py deep_ep/buffer.py +3 -3

No files found.
--- a/deep_ep/buffer.py
+++ b/deep_ep/buffer.py
@@ -308,9 +308,9 @@ class Buffer:
            8: Config(Buffer.num_sms, 6, 256, 6, 128),
            # 16: Config(Buffer.num_sms, 36, 288, 20, 128),
            16: Config(Buffer.num_sms, 8, 512, 16, 128),
-            24: Config(Buffer.num_sms, 8, 288, 32, 128),
-            32: Config(Buffer.num_sms, 32, 288, 32, 128),
-            64: Config(Buffer.num_sms, 20, 288, 28, 128),
+            24: Config(Buffer.num_sms, 8, 512, 32, 128),
+            32: Config(Buffer.num_sms, 32, 512, 32, 128),
+            64: Config(Buffer.num_sms, 20, 512, 28, 128),
            128: Config(Buffer.num_sms, 20, 560, 32, 128),
            144: Config(Buffer.num_sms, 32, 720, 12, 128),
            160: Config(Buffer.num_sms, 28, 720, 12, 128),