Skip to content

Commit d2f9ff6

Browse files
authored
Qualcomm AI Engine Direct - support XR2Gen3 SoC (#9181)
### Summary - add SXR2330P ### Test plan ```bash python backends/qualcomm/tests/test_qnn_delegate.py -k TestQNNQuantizedOperator -s $SERIAL_NO -m SM8650 -b build-android ```
1 parent f8e357a commit d2f9ff6

File tree

3 files changed

+33
-22
lines changed

3 files changed

+33
-22
lines changed

backends/qualcomm/serialization/qc_compiler_spec.fbs

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -32,16 +32,17 @@ table HtpInfo {
3232
/// to get SoC Model in supported snapdragon devices
3333
enum QcomChipset: int {
3434
UNKNOWN_SM = 0,
35+
SA8295 = 39,
3536
SM8450 = 36,
3637
SM8475 = 42,
3738
SM8550 = 43,
38-
SSG2115P = 46,
3939
SM8650 = 57,
40-
SA8295 = 39,
4140
SM8750 = 69,
41+
SSG2115P = 46,
42+
SSG2125P = 58,
4243
SXR1230P = 45,
4344
SXR2230P = 53,
44-
SSG2125P = 58,
45+
SXR2330P = 75,
4546
}
4647

4748
/// Indicate the information of the specified SoC.

backends/qualcomm/serialization/qc_schema.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -37,16 +37,17 @@ class HtpInfo:
3737
@unique
3838
class QcomChipset(IntEnum):
3939
UNKNOWN_SM = 0
40+
SA8295 = 39 # v68
4041
SM8450 = 36 # v69
4142
SM8475 = 42 # v69
4243
SM8550 = 43 # v73
43-
SSG2115P = 46 # v73
4444
SM8650 = 57 # v75
45-
SA8295 = 39 # v68
4645
SM8750 = 69 # v79
47-
SXR1230P = 45
48-
SXR2230P = 53
49-
SSG2125P = 58
46+
SSG2115P = 46 # v73
47+
SSG2125P = 58 # v73
48+
SXR1230P = 45 # v73
49+
SXR2230P = 53 # v69
50+
SXR2330P = 75 # v79
5051

5152

5253
@dataclass
@@ -56,16 +57,17 @@ class SocInfo:
5657

5758

5859
_soc_info_table = {
60+
QcomChipset.SA8295: SocInfo(QcomChipset.SA8295, HtpInfo(HtpArch.V68, 8)),
5961
QcomChipset.SM8450: SocInfo(QcomChipset.SM8450, HtpInfo(HtpArch.V69, 8)),
6062
QcomChipset.SM8475: SocInfo(QcomChipset.SM8475, HtpInfo(HtpArch.V69, 8)),
6163
QcomChipset.SM8550: SocInfo(QcomChipset.SM8550, HtpInfo(HtpArch.V73, 8)),
6264
QcomChipset.SM8650: SocInfo(QcomChipset.SM8650, HtpInfo(HtpArch.V75, 8)),
6365
QcomChipset.SM8750: SocInfo(QcomChipset.SM8750, HtpInfo(HtpArch.V79, 8)),
6466
QcomChipset.SSG2115P: SocInfo(QcomChipset.SSG2115P, HtpInfo(HtpArch.V73, 2)),
65-
QcomChipset.SA8295: SocInfo(QcomChipset.SA8295, HtpInfo(HtpArch.V68, 8)),
67+
QcomChipset.SSG2125P: SocInfo(QcomChipset.SSG2125P, HtpInfo(HtpArch.V73, 2)),
6668
QcomChipset.SXR1230P: SocInfo(QcomChipset.SXR1230P, HtpInfo(HtpArch.V73, 2)),
6769
QcomChipset.SXR2230P: SocInfo(QcomChipset.SXR2230P, HtpInfo(HtpArch.V69, 8)),
68-
QcomChipset.SSG2125P: SocInfo(QcomChipset.SSG2125P, HtpInfo(HtpArch.V73, 2)),
70+
QcomChipset.SXR2330P: SocInfo(QcomChipset.SXR2330P, HtpInfo(HtpArch.V79, 8)),
6971
}
7072

7173

backends/qualcomm/utils/utils.py

Lines changed: 20 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1280,25 +1280,33 @@ def generate_qnn_executorch_compiler_spec(
12801280

12811281
def get_soc_to_arch_map():
12821282
return {
1283-
"SSG2115P": HtpArch.V73,
1284-
"SM8750": HtpArch.V79,
1285-
"SM8650": HtpArch.V75,
1286-
"SM8550": HtpArch.V73,
1287-
"SM8475": HtpArch.V69,
1288-
"SM8450": HtpArch.V69,
12891283
"SA8295": HtpArch.V68,
1284+
"SM8450": HtpArch.V69,
1285+
"SM8475": HtpArch.V69,
1286+
"SM8550": HtpArch.V73,
1287+
"SM8650": HtpArch.V75,
1288+
"SM8750": HtpArch.V79,
1289+
"SSG2115P": HtpArch.V73,
1290+
"SSG2125P": HtpArch.V73,
1291+
"SXR1230P": HtpArch.V73,
1292+
"SXR2230P": HtpArch.V69,
1293+
"SXR2330P": HtpArch.V79,
12901294
}
12911295

12921296

12931297
def get_soc_to_chipset_map():
12941298
return {
1295-
"SSG2115P": QcomChipset.SSG2115P,
1296-
"SM8750": QcomChipset.SM8750,
1297-
"SM8650": QcomChipset.SM8650,
1298-
"SM8550": QcomChipset.SM8550,
1299-
"SM8475": QcomChipset.SM8475,
1300-
"SM8450": QcomChipset.SM8450,
13011299
"SA8295": QcomChipset.SA8295,
1300+
"SM8450": QcomChipset.SM8450,
1301+
"SM8475": QcomChipset.SM8475,
1302+
"SM8550": QcomChipset.SM8550,
1303+
"SM8650": QcomChipset.SM8650,
1304+
"SM8750": QcomChipset.SM8750,
1305+
"SSG2115P": QcomChipset.SSG2115P,
1306+
"SSG2125P": QcomChipset.SSG2125P,
1307+
"SXR1230P": QcomChipset.SXR1230P,
1308+
"SXR2230P": QcomChipset.SXR2230P,
1309+
"SXR2330P": QcomChipset.SXR2330P,
13021310
}
13031311

13041312

0 commit comments

Comments
 (0)