Skip to content

Commit 1fac477

Browse files
etafpytorchmergebot
authored andcommitted
[Break XPU][Inductor UT] Generalize device-bias code introduced by pytorch#146866. (pytorch#148534)
Pull Request resolved: pytorch#148534 Approved by: https://github.com/nandesuka
1 parent f057206 commit 1fac477

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/inductor/test_padding.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -760,7 +760,7 @@ def get_input(size: tuple[int], alignment_bytes: int) -> torch.Tensor:
760760

761761
output_shape = (shape[0] * num_inputs, shape[1])
762762
output_stride = input_tensors[0].stride()
763-
output_line = f"buf12 = empty_strided_cuda({output_shape}, {output_stride}, torch.float32)"
763+
output_line = f"buf12 = empty_strided_{GPU_TYPE}({output_shape}, {output_stride}, torch.float32)"
764764
self.assertTrue(any(output_line in line for line in code))
765765

766766

0 commit comments

Comments
 (0)