add more names

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>

Author: kylesayrs

src/llmcompressor/args/dataset_arguments.py

@@ -181,7 +181,16 @@ class DatasetArguments(CustomDatasetArguments):
         },
     )
     tracing_ignore: List[str] = field(
-        default_factory=lambda: ["_update_causal_mask", "create_causal_mask"],
+        default_factory=lambda: [
+            "_update_causal_mask",
+            "create_causal_mask",
+            "make_causal_mask",
+            "get_causal_mask",
+            "mask_function",
+            "_prepare_4d_causal_attention_mask",
+            "_prepare_fsmt_decoder_inputs",
+            "_prepare_4d_causal_attention_mask_with_cache_position",
+        ],
         metadata={
             "help": "List of functions to ignore during tracing, either "
             "{module}.{method_name} or {function_name}"