Skip to content

Commit 20e272f

Browse files
committed
Fix RUF005 (splat instead of concatenating)
1 parent 64478fd commit 20e272f

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

bitsandbytes/triton/int8_matmul_mixed_dequantize.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,8 @@ def get_configs_io_bound():
5858
triton.Config({'BLOCK_M': 64, 'BLOCK_N': 128, 'BLOCK_K': 64, 'SPLIT_K': 1}, num_stages=4, num_warps=4),
5959
triton.Config({'BLOCK_M': 128, 'BLOCK_N': 32, 'BLOCK_K': 64, 'SPLIT_K': 1}, num_stages=4, num_warps=4),
6060
triton.Config({'BLOCK_M': 64, 'BLOCK_N': 32, 'BLOCK_K': 64, 'SPLIT_K': 1}, num_stages=5, num_warps=2),
61-
] + get_configs_io_bound(),
61+
*get_configs_io_bound(),
62+
],
6263
key=['M', 'N', 'K'],
6364
prune_configs_by={
6465
'early_config_prune': early_config_prune,

bitsandbytes/triton/int8_matmul_rowwise_dequantize.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,8 @@ def get_configs_io_bound():
5757
triton.Config({'BLOCK_M': 64, 'BLOCK_N': 128, 'BLOCK_K': 64, 'SPLIT_K': 1}, num_stages=4, num_warps=4),
5858
triton.Config({'BLOCK_M': 128, 'BLOCK_N': 32, 'BLOCK_K': 64, 'SPLIT_K': 1}, num_stages=4, num_warps=4),
5959
triton.Config({'BLOCK_M': 64, 'BLOCK_N': 32, 'BLOCK_K': 64, 'SPLIT_K': 1}, num_stages=5, num_warps=2),
60-
] + get_configs_io_bound(),
60+
*get_configs_io_bound(),
61+
],
6162
key=['M', 'N', 'K'],
6263
prune_configs_by={
6364
'early_config_prune': early_config_prune,

0 commit comments

Comments
 (0)