Skip to content

Commit feba695

Browse files
committed
update docstring
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
1 parent ccb88ed commit feba695

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

src/compressed_tensors/transform/utils/hadamard.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -69,12 +69,9 @@ def random_hadamard_matrix(
6969
gen: Optional[torch.Generator] = None,
7070
) -> torch.Tensor:
7171
"""
72-
Produces a randomly generated Hadamard matrix.
73-
See https://cornell-relaxml.github.io/quip-sharp/ ,
74-
Section "Randomized Hadamard Transformation"
75-
76-
Improves upon deterministic_hadamard_matrix
77-
in that this supports non powers of 2 and random seeds
72+
Produces a randomly generated Hadamard matrix. Differs from
73+
`deterministic_hadamard_matrix` in that this function supports non powers of 2
74+
and randomization using a seeded generator
7875
7976
Adapated from https://github.com/facebookresearch/SpinQuant/blob/main/utils/hadamard_utils.py # noqa: E501
8077
Known matrices were retrieved from N. J. A. Sloane's Library of Hadamard Matrices http://www.neilsloane.com/hadamard/ # noqa: E501

0 commit comments

Comments
 (0)