diff options
| author | Grant Watson <grant.watson@arm.com> | 2023-05-31 14:56:13 +0100 |
|---|---|---|
| committer | Grant Watson <grant.watson@arm.com> | 2023-06-12 17:15:54 +0100 |
| commit | 6168047ef0354927cb175ad295722924dfc3053c (patch) | |
| tree | 033b1568c35eea8c2dc65487c48c4530a71ab2b3 /scripts | |
| parent | 5dd5a55bc00d0eaf9aa38511cf553b0d78dfed51 (diff) | |
| download | reference_model-6168047ef0354927cb175ad295722924dfc3053c.tar.gz | |
Correctly identify "axis" attributes.
- Allows axis attributes to be treated differently to other arguments in attribute.def
Signed-off-by: Grant Watson <grant.watson@arm.com>
Change-Id: I1be2595c24bf22e5391a2911a5283391d310df37
Diffstat (limited to 'scripts')
| -rw-r--r-- | scripts/operator_api/generate_api.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/scripts/operator_api/generate_api.py b/scripts/operator_api/generate_api.py index 671d902..5038973 100644 --- a/scripts/operator_api/generate_api.py +++ b/scripts/operator_api/generate_api.py @@ -192,6 +192,18 @@ def getOperators(tosaXml): operator["serializeAttType"] = getSerializeOpType(opName) tosaArgs = getTosaArgs(opXml) serializeArgs = getSerializeArgsForOp(opName, allSerializeArgs, tosaArgs) + # Handle "axis" arguments + axisList = [arg["name"] for arg in tosaArgs if arg["name"] == "axis"] + if operator["serializeAttType"] == "None" and len(axisList) > 0: + operator["serializeAttType"] = "Axis" + serializeArgs = [ + { + "name": "axis", + "dType": "int32_t", + "SV": "S", + "init": "= client_axis", + } + ] updateTosaArgs(tosaArgs, serializeArgs, tosaXml) operator["arguments"] = tosaArgs operator["serializeArgs"] = serializeArgs |