[WIP][Tracing] Mistral3ForConditionalGeneration

[kylesayrs]

Purpose

• Improve tracing logic

Prerequisites

• [Tracing] Raise _is_compiling_flag while tracing #1388
• [Tracing] Skip non-ancestors of sequential targets #1389
• [Tracing] Autowrap methods by name #1390

Changes

• Add _is_compiling_flag
• Do not trace into any modules which are not ancestors of the sequential targets
• Add ability to patch _update_causal_mask

Testing

llmcompressor.trace \
     --model_id "mistralai/Mistral-Small-3.1-24B-Instruct-2503" \
     --model_class "Mistral3ForConditionalGeneration" \
     --sequential_targets "MistralDecoderLayer" \
     --ignore "MistralModel._update_causal_mask" \
     --modality vision \

[github-actions]

👋 Hi! Thank you for contributing to llm-compressor. Please add the ready label when the PR is ready for review.

Note: This is required to complete the testing suite, please only add the label once the PR is code complete and local testing has been performed.