From 1b66ea9567dfc2995f954ecd8e887ab49406404d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?M=2E=20Tolga=20Cang=C3=B6z?= Date: Wed, 14 Feb 2024 13:59:03 +0300 Subject: [PATCH] Add text type --- quanto/library/ext/README.md | 2 +- quanto/library/python/README.md | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/quanto/library/ext/README.md b/quanto/library/ext/README.md index 8009519c..f2b59a97 100644 --- a/quanto/library/ext/README.md +++ b/quanto/library/ext/README.md @@ -14,7 +14,7 @@ The operations are defined in `library/ops.py`. To provide an implementation for specific device types, use the following syntax: -``` +```python @torch.library.impl("quanto_ext::unpack", ["CPU", "CUDA"]) def unpack(packed: torch.Tensor, bits: int) -> torch.Tensor: return ext().unpack(t, bits) diff --git a/quanto/library/python/README.md b/quanto/library/python/README.md index bc270fc0..8c33eda0 100644 --- a/quanto/library/python/README.md +++ b/quanto/library/python/README.md @@ -8,7 +8,7 @@ The operations are defined in `library/ops.py`. To provide an implementation for an operation, use the following syntax: -``` +```python @torch.library.impl("quanto_py::unpack", "default") def unpack(packed: torch.Tensor, bits: int) -> torch.Tensor: ...