Skip to content

Commit 085dd96

Browse files
committed
update doc
1 parent 980c623 commit 085dd96

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

Readme.md

+1
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,7 @@ HQQLinear.set_backend(HQQBackend.PYTORCH_COMPILE) #Compiled Pytorch via dynamo
5151
HQQLinear.set_backend(HQQBackend.ATEN) #C++ Aten/CUDA backend (set automatically by default if available)
5252
```
5353
The ```HQQBackend.ATEN``` backend is automatically installed and used by default when available.
54+
Note that ```HQQBackend.ATEN``` only supports `axis=0`. For `axis=1` you ned to use ```HQQBackend.PYTORCH``` or ```HQQBackend.PYTORCH_COMPILE```.
5455

5556
Below you can find the speed-up benchmark with various backends, ```HQQBackend.PYTORCH``` being the baseline:
5657

0 commit comments

Comments
 (0)