[Transform] [Utils] Canonical matching utilities

[kylesayrs]

Purpose

• There currently exist 23 different functions for matching modules/parameters. These utils consolidate them to just 6.
• Used by transforms for module matching and layer norm matching

Function	Used for/by	Maps to
get_layer_by_name	awq	attrgetter(layer_name)(module)
get_layers_params	pruning modifiers	match_named_parameters
get_param	None	Delete
get_params	get_layers_params , get_param	Delete
set_layer	Distillation modifier	Module.set_submodule
get_layer	None	Delete
get_layers	Lots	match_named_modules
match_layers_params	get_layers	Delete
get_default_params	match_layers_params	Delete
match_targets	Many	match_named_modules
get_quantizable_layers	match_layers_params	Delete, use explicit targets
get_prunable_layers	match_layers_params	Delete, use explicit targets
get_terminal_layers	match_layers_params	Delete, use explicit targets
get_matching_layer	Smoothquant	match_modules_set
AWQ.set_resolved_mappings	AWQ	match_modules_set
expand_target_names	Compressor	match_named_parameters
is_target	Quantization	is_match
find_name_or_class_matches	Quantization	match_name or match_class
_find_matches	Quantization	Delete
get_nested_weight_mappings	Compressor	match_named_parameters
match_param_name	get_nested_weight_mappings	Delete,match_name
match_modules	sequential pipeline	match_named_modules
get_linear_layers	infer_sparsity_structure_from_model	match_named_modules

get_linear_layers

Changes

• Implement 3 commonly-used functions
  • match_named_modules returns modules
  • match_named_parameters returns parameters (and their parent modules)
  • match_modules_set returns modules in predefined groups
• Implements 3 helper functions which are publicly exposed
  • match_name returns True if target regex matches the name
  • match_class returns True if target matches any parent classes
  • is_match returns True if either match_name or match_class is True

Integration Plan

• This PR utilizes these matching utilities for transforms. Future PRs will replace existing functions with these function as described in the table

Testing

• Added comprehensive tests generated by Claude, which I have manually reviewed and fixed any issues with

[dsikka]

Do we need llm-comrpessor or vLLM changes or any tests for backwards compatibility?
Should we test this with the nightly given how many things we're touching here?

[shanjiaz]

Thanks for adding tests! Only thing is that I see a lot of repeated logic across match_named_modules, match_named_parameters, and match_modules_set. Would be great if they can be refactored. Otherwise looks good to me!

[kylesayrs]

Confirmed that vLLM does not use any CT matching utilities
https://github.com/search?q=repo%3Avllm-project%2Fvllm+%22from+compressed_tensors%22&type=code

[kylesayrs]

@dsikka

Do we need llm-comrpessor or vLLM changes or any tests for backwards compatibility?
Should we test this with the nightly given how many things we're touching here?

Not right now, since this PR only adds new utilities. The other utilities will be removed in follow ups. Once this lands I can create an issue which users can contribute to