Re-quantize DeepSeek model weights to support DeepGEMM new input format (#7156)

2025-06-13 15:57:45 -07:00
parent c49c1d9226
commit 5b1afa7814
3 changed files with 125 additions and 0 deletions
@@ -0,0 +1,8 @@
+# COPIED FROM DeepGEMM
+def align(x: int, y: int) -> int:
+    return ceil_div(x, y) * y
+
+
+# COPIED FROM DeepGEMM
+def ceil_div(x: int, y: int) -> int:
+    return (x + y - 1) // y