Skip to content

Commit 28c603c

Browse files
committed
[Fix]: Adding support for symmetric_no_clipping_error on latest main
1 parent 5873032 commit 28c603c

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

LICENSE

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,6 @@
11
Copyright 2023 Meta
2+
All contributions by Arm:
3+
Copyright (c) 2024-2025 Arm Limited and/or its affiliates
24

35
Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
46

torchao/quantization/quant_api.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# Copyright (c) Meta Platforms, Inc. and affiliates.
2+
# Copyright 2024-2025 Arm Limited and affiliates.
23
# All rights reserved.
3-
44
# This source code is licensed under the license found in the
55
# LICENSE file in the root directory of this source tree.
66

@@ -862,8 +862,9 @@ def __post_init__(self):
862862
assert self.weight_mapping_type in [
863863
MappingType.ASYMMETRIC,
864864
MappingType.SYMMETRIC,
865+
MappingType.SYMMETRIC_NO_CLIPPING_ERR,
865866
], (
866-
f"weight_mapping_type must be MappingType.ASYMMETRIC or MappingType.SYMMETRIC, but got {self.weight_mapping_type}"
867+
f"weight_mapping_type must be MappingType.ASYMMETRIC or MappingType.SYMMETRIC or MappingType.SYMMETRIC_NO_CLIPPING_ERR, but got {self.weight_mapping_type}"
867868
)
868869
assert self.act_mapping_type in [
869870
MappingType.ASYMMETRIC,

0 commit comments

Comments
 (0)