Skip to content

Commit

Permalink
[sharktank] Mark test expected to fail
Browse files Browse the repository at this point in the history
Marks `testExportNondecomposed` as expected to fail if
running with torch>=2.4.0, see nod-ai#684.
  • Loading branch information
marbre committed Dec 12, 2024
1 parent d5cf62b commit 0af8d0c
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions sharktank/tests/layers/paged_llama_attention_block_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@
# See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

import pytest

import logging

logging.basicConfig(level=logging.DEBUG)
Expand Down Expand Up @@ -118,6 +120,10 @@ def forward(self, h, seq_block_ids, cache_state):
asm = str(output.mlir_module)
self.assertNotIn("scaled_dot_product_attention", asm)

@pytest.mark.xfail(
torch.__version__ >= (2, 4),
reason="https://github.com/nod-ai/shark-ai/issues/684",
)
def testExportNondecomposed(self):
dtype = torch.float32

Expand Down

0 comments on commit 0af8d0c

Please sign in to comment.