Skip to content

Commit 7ef8e57

Browse files
oscarandersson8218perheld
authored andcommitted
Arm backend: Update Vela to 4.4.0 python package
Co-authored-by: Per Held <[email protected]> Change-Id: I62e3856bfdee08926d6861a4ef1f5213dafaa231 Signed-off-by: Zingo Andersen <[email protected]>
1 parent 59534ac commit 7ef8e57

File tree

5 files changed

+8
-30
lines changed

5 files changed

+8
-30
lines changed

backends/arm/requirements-arm-ethos-u.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,4 @@
33
# This source code is licensed under the BSD-style license found in the
44
# LICENSE file in the root directory of this source tree.
55

6-
ethos-u-vela @ git+https://gitlab.arm.com/artificial-intelligence/ethos-u/ethos-u-vela@9a43a1bf26bfc7588358d7e6e6bb2613b4981a34
6+
ethos-u-vela == 4.4.0

backends/arm/test/ops/test_logical.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -86,9 +86,6 @@ def forward(self, tensor: torch.Tensor):
8686
#################
8787

8888

89-
xfails = {"rand_rank4": "MLBEDSW-11031: Output diff on u85 bool transpose."}
90-
91-
9289
@common.parametrize("test_data", And().test_data)
9390
def test_logical_and_tosa_FP(test_data: input_t2):
9491
pipeline = TosaPipelineFP[input_t2](
@@ -132,7 +129,7 @@ def test_logical_and_u55_INT_not_delegated(test_data: input_t2):
132129
pipeline.run()
133130

134131

135-
@common.parametrize("test_data", And().test_data, xfails=xfails)
132+
@common.parametrize("test_data", And().test_data)
136133
@common.XfailIfNoCorstone320
137134
def test_logical_and_u85_INT(test_data: input_t2):
138135
pipeline = EthosU85PipelineINT[input_t2](
@@ -226,7 +223,7 @@ def test_logical_xor_u55_INT_not_delegated(test_data: input_t2):
226223
pipeline.run()
227224

228225

229-
@common.parametrize("test_data", Xor().test_data, xfails=xfails)
226+
@common.parametrize("test_data", Xor().test_data)
230227
@common.XfailIfNoCorstone320
231228
def test_logical_xor_u85_INT(test_data: input_t2):
232229
pipeline = EthosU85PipelineINT[input_t2](
@@ -320,7 +317,7 @@ def test_logical_or_u55_INT_not_delegated(test_data: input_t2):
320317
pipeline.run()
321318

322319

323-
@common.parametrize("test_data", Or().test_data, xfails=xfails)
320+
@common.parametrize("test_data", Or().test_data)
324321
@common.XfailIfNoCorstone320
325322
def test_logical_or_u85_INT(test_data: input_t2):
326323
pipeline = EthosU85PipelineINT[input_t2](
@@ -414,7 +411,7 @@ def test_logical_not_u55_INT_not_delegated(test_data: input_t2):
414411
pipeline.run()
415412

416413

417-
@common.parametrize("test_data", Not().test_data, xfails=xfails)
414+
@common.parametrize("test_data", Not().test_data)
418415
@common.XfailIfNoCorstone320
419416
def test_logical_not_u85_INT(test_data: input_t2):
420417
pipeline = EthosU85PipelineINT[input_t2](

backends/arm/test/ops/test_logsoftmax.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -64,13 +64,7 @@ def test_log_softmax_tosa_INT(test_data):
6464
pipeline.run()
6565

6666

67-
@common.parametrize(
68-
"test_data",
69-
LogSoftmax.test_data,
70-
xfails={
71-
"randn_neg_dim": "MLBEDSW-11032: ILLEGAL_OFM_BASE error: Base addresses must be aligned to brick depth on u55."
72-
},
73-
)
67+
@common.parametrize("test_data", LogSoftmax.test_data)
7468
@common.XfailIfNoCorstone300()
7569
def test_log_softmax_u55_INT(test_data):
7670
data, dim = test_data()

backends/arm/test/ops/test_softmax.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -61,13 +61,7 @@ def test_softmax_tosa_INT(test_data):
6161
pipeline.run()
6262

6363

64-
@common.parametrize(
65-
"test_data",
66-
Softmax.test_data,
67-
{
68-
"randn_neg_dim": "MLBEDSW-11032: ILLEGAL_OFM_BASE error: Base addresses must be aligned to brick depth on u55."
69-
},
70-
)
64+
@common.parametrize("test_data", Softmax.test_data)
7165
@common.XfailIfNoCorstone300
7266
def test_softmax_u55_INT(test_data):
7367
data, dim = test_data()

backends/arm/test/passes/test_rescale_pass.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -172,14 +172,7 @@ def test_quantized_rescale_tosa_bi(test_data: tuple[torch.Tensor, torch.Tensor])
172172
pipeline.run()
173173

174174

175-
u55_xfails = {
176-
"ones": "MLBEDSW-11032: ILLEGAL_OFM_BASE error: Base addresses must be aligned to brick depth on u55.",
177-
"randn_ones": "MLBEDSW-11032: ILLEGAL_OFM_BASE error: Base addresses must be aligned to brick depth on u55.",
178-
"randn_large": "MLBEDSW-11032: ILLEGAL_OFM_BASE error: Base addresses must be aligned to brick depth on u55.",
179-
}
180-
181-
182-
@common.parametrize("test_data", RescaleNetwork.test_data, xfails=u55_xfails)
175+
@common.parametrize("test_data", RescaleNetwork.test_data)
183176
@common.XfailIfNoCorstone300
184177
def test_quantized_rescale_u55(test_data: tuple[torch.Tensor, torch.Tensor]):
185178
"""Tests a model with many ops that requires rescales. As more ops are quantized to int32 and

0 commit comments

Comments
 (0)