diff --git a/backends/qualcomm/serialization/qnn_compile_spec_schema.py b/backends/qualcomm/serialization/qnn_compile_spec_schema.py index 8471aad982d..09d910aba58 100644 --- a/backends/qualcomm/serialization/qnn_compile_spec_schema.py +++ b/backends/qualcomm/serialization/qnn_compile_spec_schema.py @@ -33,6 +33,7 @@ class QcomChipset(IntEnum): SM8450 = 36 # v69 SM8475 = 42 # v69 SM8550 = 43 # v73 + SSG2115P = 46 # v73 SM8650 = 57 # v75 @@ -47,6 +48,7 @@ class SocInfo: QcomChipset.SM8475: SocInfo(QcomChipset.SM8475, HtpInfo(HtpArch.V69, 8)), QcomChipset.SM8550: SocInfo(QcomChipset.SM8550, HtpInfo(HtpArch.V73, 8)), QcomChipset.SM8650: SocInfo(QcomChipset.SM8650, HtpInfo(HtpArch.V75, 8)), + QcomChipset.SSG2115P: SocInfo(QcomChipset.SSG2115P, HtpInfo(HtpArch.V73, 2)), } diff --git a/backends/qualcomm/serialization/schema.fbs b/backends/qualcomm/serialization/schema.fbs index 4e7fdb56e89..f2275377f7b 100644 --- a/backends/qualcomm/serialization/schema.fbs +++ b/backends/qualcomm/serialization/schema.fbs @@ -32,6 +32,7 @@ enum QcomChipset: int { SM8450 = 36, SM8475 = 42, SM8550 = 43, + SSG2115P = 46, SM8650 = 57, } @@ -170,7 +171,7 @@ table QnnExecuTorchOptions { /// Profiling level of the delegate and the backend. Default is off. profile_level:QnnExecuTorchProfileLevel; - + /// Enables usage of shared buffer between application and backend for graph I/O. shared_buffer:bool; diff --git a/backends/qualcomm/tests/utils.py b/backends/qualcomm/tests/utils.py index 7209b0a2678..52ffac46eee 100644 --- a/backends/qualcomm/tests/utils.py +++ b/backends/qualcomm/tests/utils.py @@ -118,6 +118,7 @@ class TestQNN(unittest.TestCase): model: QcomChipset = None compiler_specs: List[CompileSpec] = None arch_table = { + "SSG2115P": QcomChipset.SSG2115P, "SM8650": QcomChipset.SM8650, "SM8550": QcomChipset.SM8550, "SM8475": QcomChipset.SM8475, diff --git a/examples/qualcomm/oss_scripts/llama2/llama.py b/examples/qualcomm/oss_scripts/llama2/llama.py index d74cfa0ef07..f16736248dd 100644 --- a/examples/qualcomm/oss_scripts/llama2/llama.py +++ b/examples/qualcomm/oss_scripts/llama2/llama.py @@ -48,6 +48,7 @@ soc_to_chipset_map = { + "SSG2115P": QcomChipset.SSG2115P, "SM8650": QcomChipset.SM8650, "SM8550": QcomChipset.SM8550, "SM8475": QcomChipset.SM8475, diff --git a/examples/qualcomm/utils.py b/examples/qualcomm/utils.py index 9c4cd4453f0..ac16343a8e8 100755 --- a/examples/qualcomm/utils.py +++ b/examples/qualcomm/utils.py @@ -83,6 +83,7 @@ def __init__( self.debug_output_path = f"{self.workspace}/debug_output.bin" self.output_folder = f"{self.workspace}/outputs" self.arch_table = { + "SSG2115P": "73", "SM8650": "75", "SM8550": "73", "SM8475": "69",