Skip to content

Commit ecd2a86

Browse files
committed
Fix CI
1 parent a11e455 commit ecd2a86

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

test/test_ops.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -271,6 +271,8 @@ def test_dequantize_tensor_core_tiled_layout_op(shape, inner_k_tiles, group_size
271271
device = "cuda"
272272

273273
q = torch.randint(0, 16, shape, dtype=torch.int, device=device)
274+
if TORCH_VERSION_AFTER_2_5:
275+
q = (q[::, ::2] << 4 | q[::, 1::2]).to(torch.uint8)
274276
packed_w = torch._convert_weight_to_int4pack(q, inner_k_tiles)
275277
q_groups = k // group_size
276278
scales = torch.randn(n, q_groups, dtype=torch.bfloat16, device=device)

0 commit comments

Comments
 (0)