Skip to content

Commit 3c3c64a

Browse files
authored
Fix test_inline_asm_packed expected output due to upstream PyTorch change (#385)
1 parent 2a9aa70 commit 3c3c64a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_inline_asm_elementwise.expected

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def _kernel_packed_asm_kernel(x, result, x_size_0, result_stride_0, x_stride_0,
9292
mask_0 = indices_0 < x_size_0
9393
val = tl.load(x + indices_0 * x_stride_0, mask_0, other=0)
9494
result_val = tl.inline_asm_elementwise('and.b32 $0, $1, 0x1F1F1F1F; shl.b32 $0, $0, 3;', '=r,r', [val], tl.int8, True, 4)
95-
v_0 = result_val.to(tl.int8).to(tl.uint8)
95+
v_0 = result_val.to(tl.uint8)
9696
tl.store(result + indices_0 * result_stride_0, v_0, mask_0)
9797

9898
def kernel_packed_asm(x: torch.Tensor, *, _launcher=_default_launcher):

0 commit comments

Comments
 (0)