Skip to content

Commit fc6d097

Browse files
authored
Disable llm module test (#6861)
disable llm module test
1 parent 71612a6 commit fc6d097

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

extension/llm/modules/test/test_attention.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -146,6 +146,7 @@ def test_attention_export(self):
146146

147147
assert_close(et_res, tt_res)
148148

149+
@unittest.skip(reason="TODO(T207740932): test is flaky")
149150
def test_attention_aoti(self):
150151
# Self attention.
151152

extension/llm/modules/test/test_position_embeddings.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -163,6 +163,7 @@ def test_tiled_token_positional_embedding_export(self):
163163

164164
assert_close(y, ref_y)
165165

166+
@unittest.skip(reason="TODO(T207740932): test is flaky")
166167
def test_tiled_token_positional_embedding_aoti(self):
167168
tpe_ep = torch.export.export(
168169
self.tpe,

0 commit comments

Comments
 (0)