Skip to content

Commit 1823af4

Browse files
committed
Merge branch 'kylesayrs/extend-hadamard', remote-tracking branch 'origin' into kylesayrs/transform_construct_cache_device
3 parents b715329 + adae77b + 249323c commit 1823af4

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

src/compressed_tensors/transform/factory/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ def _apply_to_module(self, module: Module, args: TransformArgs):
9797
:param args: defines how the transform will be applied to the target module
9898
"""
9999
# create transform as submodule
100-
transform_name = f"{self.name}_{args.location}"
100+
transform_name = f"{self.name}_{args.location.value}"
101101
transform = self.create_transform(module, args)
102102
register_offload_module(module, transform_name, transform) # (1)
103103

src/compressed_tensors/transform/utils/hadamard.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,7 @@ def random_hadamard_matrix(
7575
in that this supports non powers of 2 and random seeds
7676
7777
Adapated from https://github.com/facebookresearch/SpinQuant/blob/main/utils/hadamard_utils.py # noqa: E501
78+
Known matrices were retrieved from N. J. A. Sloane's Library of Hadamard Matrices http://www.neilsloane.com/hadamard/ # noqa: E501
7879
7980
:param size: The dimension of the hamadard matrix
8081
:param gen: Optional generator random values

0 commit comments

Comments
 (0)