utils
=====

.. List the submodules





.. Autodoc anything defined in the module itself

   TODO: WE DON'T USE THIS OPTION RIGHT NOW BUT WE CAN REACTIVATE IF WANTED
   We use :ignore-module-all: so sphinx does not document the same module twice, even if it is reimported
   For reimports that should be documented somewhere other than where they are defined, the re-imports
   __module__ should be manually overridden -- i.e. in the ``__init__.py`` which contains ``from xxx import YYY``,
   add in ``YYY.__module__ = __name__``.

.. automodule:: modelopt.torch.quantization.utils
   :members:
   :undoc-members:

   .. Also show members without docstrings. Only members from __all__ are considered as per conf.py
   .. Ideally we should add docstrings for these members.


   .. Overview table of available classes in the module
   
   
   .. rubric:: Classes

   .. autosummary::
      :nosignatures:
   
      SharedQuantState
      SharedWeightGlobalAmaxState
   
   


   .. Overview table of available functions in the module
   
   
   .. rubric:: Functions

   .. autosummary::
      :nosignatures:
   
      convert_quantization_axis_to_reduce_axis
      export_torch_mode
      find_shared_input_groups
      is_quantized
      is_quantized_column_parallel_linear
      is_quantized_linear
      is_quantized_row_parallel_linear
      iter_shared_quant_states
      reduce_amax
      reduce_sum
      replace_function
      representative_weight_quantizer
      update_quant_cfg_with_kv_cache_quant
      weight_attr_names