Skip to content

Commit 54c5fab

Browse files
committed
Arm backend: Adjust tests
Signed-off-by: Erik Lundell <erik.lundell@arm.com> Change-Id: I27bd905a66c5d8027a1a50a7cf662e0851bddde7
1 parent cafbb3a commit 54c5fab

2 files changed

Lines changed: 0 additions & 10 deletions

File tree

backends/arm/test/models/test_llama.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -245,12 +245,6 @@ def test_llama_tosa_INT_static():
245245
use_to_edge_transform_and_lower=True,
246246
fold_quantize=True,
247247
)
248-
# NOTE: HF StaticCache INT currently keeps two delegated subgraphs
249-
# after partitioning on this path, so expect two delegate calls in EXIR.
250-
pipeline.change_args(
251-
"check_count.exir",
252-
{"torch.ops.higher_order.executorch_call_delegate": 2},
253-
)
254248
pipeline.run()
255249

256250

backends/arm/test/ops/test_to_copy.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -297,10 +297,6 @@ def test_to_vgf_no_quant(test_data: Tuple):
297297
in ToCopySupported::is_node_tosa_supported() before it goes into the delegated graph.
298298
"""
299299
_TO_COPY_TEST_DATA_INT = {
300-
"rand_int32_fp32": lambda: (
301-
torch.randint(-127, 128, (1, 2, 3, 4), dtype=torch.int32),
302-
torch.float32,
303-
),
304300
"rand_int32_fp16": lambda: (
305301
torch.randint(-127, 128, (1, 2, 3, 4), dtype=torch.int32),
306302
torch.float16,

0 commit comments

Comments
 (0)