make mxtensor printing nicer

[vkuzo]

Summary:

Fix printing of linear weight wrapped with MXTensor.

Test Plan:

quantize a Qwen MoE model with mxfp4 and print it

old version would print the data for each weight

new version prints this:

        (self_attn): Qwen2MoeSdpaAttention(
          (q_proj): Linear(in_features=2048, out_features=2048, weight=MXTensor(self._elem_dtype=torch.float4_e2m1fn_x2, self._block_size=32, torch.bfloat16, MXGemmKernelChoice.EMULATED,
self.act_quant_kwargs=QuantizeTensorToMXKwargs(elem_dtype=torch.float4_e2m1fn_x2, block_size=32, scaling_mode=<ScaleCalculationMode.FLOOR: 'floor'>, gemm_kernel_choice=<MXGemmKernelChoice.EMULATED: 'emulated'>,
pack_fp6=False)))

Reviewers:

Subscribers:

Tasks:

Tags:

[vkuzo]

Stack from ghstack (oldest at bottom):

• enroll mxtensor in vllm integration tests #3081
• mxtensor: switch to AOBaseTensor dispatch #3080
• mxtensor: add index select support #3079
• mxtensor: add serialization support #3078
• mxtensor: delete to_copy override #3072
• mxtensor: support clone #3070
• -> make mxtensor printing nicer #3068

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/3068

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

This comment was automatically generated by Dr. CI and updates every 15 minutes.