add mode parameter for quantization

2025-12-16 01:49:05 +08:00 · 2025-08-15 17:36:55 -07:00
parent 584d48458e
commit e04e17e3b6
9 changed files with 127 additions and 56 deletions
--- a/docs/src/dev/custom_metal_kernels.rst
+++ b/docs/src/dev/custom_metal_kernels.rst
@@ -127,7 +127,7 @@ relying on a copy from ``ensure_row_contiguous``:
      name="myexp_strided",
      input_names=["inp"],
      output_names=["out"],
-      source=source
+      source=source,
      ensure_row_contiguous=False,
  )