Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions backends/arm/test/misc/test_debug_feats.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@
TosaPipelineFP,
TosaPipelineINT,
)
from executorch.backends.test.harness.stages import StageType


input_t1 = Tuple[torch.Tensor] # Input x
Expand Down Expand Up @@ -104,7 +105,7 @@ def test_INT_artifact(test_data: input_t1):

@common.parametrize("test_data", Linear.inputs)
def test_numerical_diff_print(test_data: input_t1):
pipeline = TosaPipelineFP[input_t1](
pipeline = TosaPipelineINT[input_t1](
Linear(),
test_data,
[],
Expand All @@ -119,7 +120,7 @@ def test_numerical_diff_print(test_data: input_t1):
# not present.
try:
# Tolerate 0 difference => we want to trigger a numerical diff
tester.run_method_and_compare_outputs(atol=0, rtol=0, qtol=0)
tester.run_method_and_compare_outputs(stage=StageType.INITIAL_MODEL, atol=0, rtol=0, qtol=0)
except AssertionError:
pass # Implicit pass test
else:
Expand Down
Loading