0

[inductor] Update Triton CPU pin ()

This incorporates the fix in
https://github.com/triton-lang/triton/pull/4871.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/137778
Approved by: https://github.com/Skylion007
This commit is contained in:
Jez Ng
2024-10-11 09:44:20 -07:00
committed by PyTorch MergeBot
parent d52b2cf92f
commit c8609cf4b0
2 changed files with 1 additions and 5 deletions
.ci/docker/ci_commit_pins
test/inductor

@@ -1 +1 @@
6a333f1b05671f6fada4ba7bbfae4a02a9d96f4f
c7711371cace304afe265c1ffa906415ab82fc66

@@ -1487,7 +1487,6 @@ class CommonTemplate:
actual = nested_opt(*example_inputs)
self.assertEqual(expect, actual)
@xfail_if_triton_cpu
def test_index_propagation_flip(self):
def flip(x):
i = torch.arange(x.size(0) - 1, -1, -1, device=x.device)
@@ -1669,7 +1668,6 @@ class CommonTemplate:
vectorize=False, # There's no loop to vectorize!
)
@xfail_if_triton_cpu
def test_computed_buffer_inlining(self):
def flip(x):
idx = torch.arange(x.size(0) - 1, -1, -1, device=x.device)
@@ -10786,7 +10784,6 @@ class CommonTemplate:
self.common(fn, (inp,), check_lowp=False)
@requires_gpu()
@xfail_if_triton_cpu
@config.patch(implicit_fallbacks=True)
def test_mutable_custom_op_fixed_layout2(self):
with torch.library._scoped_library("mylib", "DEF") as lib:
@@ -11541,7 +11538,6 @@ class CommonTemplate:
self.common(forward, ())
@xfail_if_triton_cpu
def test_flip_cat(self):
def forward(unsqueeze, unsqueeze_1):
cat_1 = torch.ops.aten.cat.default([unsqueeze, unsqueeze_1], 1)