Add new tests: 4b dq, kv sdpa (#2378)

Summary:

Add new tests: 4b dq, kv sdpa

Reviewed By: manuelcandales

Differential Revision: D54806721

Author: Michael Gschwind

examples/models/llama2/export_llama_lib.py

@@ -476,13 +476,16 @@ def _export_llama(modelname, args) -> str:  # noqa: C901
     if builder.dtype == DType.fp16:
         modelname = f"{modelname}_h"
 
-    builder.save_to_pte(modelname)
-
     if args.output_name:
         modelname = args.output_name
         if modelname[-4:] == ".pte":
             modelname = modelname[:-4]
+            output_file = modelname
+        else:
+            output_file = f"{builder.output_dir}/{modelname}.pte"
+    else:
+        output_file = f"{builder.output_dir}/{modelname}.pte"
 
-    output_file = f"{builder.output_dir}/{modelname}.pte"
+    builder.save_to_pte(output_file)
 
     return output_file