Skip to content

Commit

Permalink
[Inductor] Skip test_nll_loss_backward for intel GPU. (#126157)
Browse files Browse the repository at this point in the history
Skip this test case due to unaligned behavior to CUDA for Triton `mask_load`. We submitted issue #126173 to elaborate on the root cause. We intend to skip this case for XPU first as we need to take some time to fix the issue and have full validation to update the Triton commit pin for Intel GPU.

Pull Request resolved: #126157
Approved by: https://github.com/EikanWang, https://github.com/peterbell10, https://github.com/desertfire
  • Loading branch information
etaf authored and ZelboK committed May 19, 2024
1 parent 074173b commit d0688dd
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions test/inductor/test_torchinductor.py
Original file line number Diff line number Diff line change
Expand Up @@ -6126,6 +6126,7 @@ def fn(a, b):
(a, b),
)

@skipIfXpu
def test_nll_loss_backward(self):
def fn(a, b, c):
return aten.nll_loss_backward(
Expand Down

0 comments on commit d0688dd

Please sign in to comment.