Skip to content

GitLab

Commit 8f9cb191 authored by lishen's avatar lishen
Browse files

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

parent 8c44b1fa
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
deep_ep/buffer.py
View file @ 8f9cb191
...@@ -308,9 +308,9 @@ class Buffer: ...@@ -308,9 +308,9 @@ class Buffer:
8: Config(Buffer.num_sms, 6, 256, 6, 128), 8: Config(Buffer.num_sms, 6, 256, 6, 128),
# 16: Config(Buffer.num_sms, 36, 288, 20, 128), # 16: Config(Buffer.num_sms, 36, 288, 20, 128),
16: Config(Buffer.num_sms, 8, 512, 16, 128), 16: Config(Buffer.num_sms, 8, 512, 16, 128),
24: Config(Buffer.num_sms, 8, 288, 32, 128), 24: Config(Buffer.num_sms, 8, 512, 32, 128),
32: Config(Buffer.num_sms, 32, 288, 32, 128), 32: Config(Buffer.num_sms, 32, 512, 32, 128),
64: Config(Buffer.num_sms, 20, 288, 28, 128), 64: Config(Buffer.num_sms, 20, 512, 28, 128),
128: Config(Buffer.num_sms, 20, 560, 32, 128), 128: Config(Buffer.num_sms, 20, 560, 32, 128),
144: Config(Buffer.num_sms, 32, 720, 12, 128), 144: Config(Buffer.num_sms, 32, 720, 12, 128),
160: Config(Buffer.num_sms, 28, 720, 12, 128), 160: Config(Buffer.num_sms, 28, 720, 12, 128),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment