diff --git a/README.md b/README.md index 381adea..d3cbbdf 100644 --- a/README.md +++ b/README.md @@ -8,7 +8,7 @@ Despite its lightweight design, DeepGEMM's performance matches or exceeds expert ## Performance -We test all shapes potentially used in DeepSeek-V3/R1 inference (including both prefilling and decoding, but without tensor parallelism) on H800 with NVCC 12.8. All speedup metrics are calculated in comparison to our internally and carefully optimized implementation based on CUTLASS 3.6. +We test all shapes potentially used in DeepSeek-V3/R1 inference (including both prefilling and decoding, but without tensor parallelism) on H800 SXM5 with NVCC 12.8. All speedup metrics are calculated in comparison to our internally and carefully optimized implementation based on CUTLASS 3.6. DeepGEMM does not behave very well on some shapes, optimization PRs are welcomed if you are interested.