address review

Author: shewu-quic

examples/models/llama2/export_llama_lib.py

@@ -319,7 +319,7 @@ def build_args_parser() -> argparse.ArgumentParser:
 
     parser.add_argument(
         "--soc_model",
-        help="SoC model of current device. e.g. 'SM8650' for Snapdragon 8 Gen 3.",
+        help="[QNN backend] SoC model of current device. e.g. 'SM8650' for Snapdragon 8 Gen 3.",
         type=str,
         required=False,
         default="SM8650",

examples/qualcomm/utils.py

@@ -230,19 +230,12 @@ def build_executorch_binary(
     else:
         edge_prog = capture_program(model, inputs)
 
-    arch_table = {
-        "SM8650": QcomChipset.SM8650,
-        "SM8550": QcomChipset.SM8550,
-        "SM8475": QcomChipset.SM8475,
-        "SM8450": QcomChipset.SM8450,
-    }
-
     backend_options = generate_htp_compiler_spec(
         use_fp16=False if quant_dtype else True
     )
     qnn_partitioner = QnnPartitioner(
         generate_qnn_executorch_compiler_spec(
-            soc_model=arch_table[soc_model],
+            soc_model=getattr(QcomChipset, soc_model),
             backend_options=backend_options,
             debug=False,
             saver=False,