Skip to content

Commit bac5d4a

Browse files
github-actions[bot]yanbing-j
authored andcommitted
Apply automatic Ruff fixes
1 parent 928addc commit bac5d4a

File tree

4 files changed

+10
-5
lines changed

4 files changed

+10
-5
lines changed

torchao/dtypes/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,12 +16,12 @@
1616
from .nf4tensor import NF4Tensor, to_nf4
1717
from .uintx import (
1818
BlockSparseLayout,
19+
Int4CPULayout,
1920
MarlinQQQLayout,
2021
MarlinSparseLayout,
2122
SemiSparseLayout,
2223
TensorCoreTiledLayout,
2324
UintxLayout,
24-
Int4CPULayout,
2525
)
2626
from .utils import (
2727
Layout,

torchao/dtypes/uintx/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,8 @@
1111
SemiSparseLayout,
1212
)
1313
from .tensor_core_tiled_layout import (
14-
TensorCoreTiledLayout,
1514
Int4CPULayout,
15+
TensorCoreTiledLayout,
1616
)
1717
from .uintx_layout import (
1818
UintxLayout,

torchao/dtypes/uintx/tensor_core_tiled_layout.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,12 @@
1313
)
1414
from torchao.dtypes.utils import AQTTensorImpl, Layout, is_device
1515
from torchao.quantization.quant_primitives import ZeroPointDomain, _get_reduction_params
16-
from torchao.utils import TORCH_VERSION_AT_LEAST_2_5, TORCH_VERSION_AT_LEAST_2_6, fill_defaults, find_multiple
16+
from torchao.utils import (
17+
TORCH_VERSION_AT_LEAST_2_5,
18+
TORCH_VERSION_AT_LEAST_2_6,
19+
fill_defaults,
20+
find_multiple,
21+
)
1722

1823
aten = torch.ops.aten
1924

torchao/quantization/GPTQ.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -616,8 +616,8 @@ def __init__(
616616
),
617617
dtype=torch.uint8,
618618
device=device,
619-
),
620-
)
619+
),
620+
)
621621
else:
622622
self.register_buffer(
623623
"weight",

0 commit comments

Comments
 (0)