In tutorials/quantize_vit, extract common methods to util.py

[lancerts]

In tutorials/quantize_vit, extract common methods to util.py

Local execution tested via ./run.sh

cc @msaroufim

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

✅ No Failures

As of commit b7d6811 with merge base adfe570 ():
💚 Looks good so far! There are no failures yet. 💚

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

[lancerts]

 =================================== FAILURES ===================================
  _____ TestWeightOnlyInt8Quant.test_weight_only_quant_force_mixed_mm_4_cuda _____
  
  a = (<test_integration.TestWeightOnlyInt8Quant testMethod=test_weight_only_quant_force_mixed_mm_4_cuda>,)
  kw = {}
  
      @wraps(func)
      def standalone_func(*a, **kw):
  >       return func(*(a + p.args), **p.kwargs, **kw)
  
  /opt/conda/envs/venv/lib/python3.9/site-packages/parameterized/parameterized.py:620: 
  _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
  /opt/conda/envs/venv/lib/python3.9/site-packages/torch/utils/_contextlib.py:115: in decorate_context
      return func(*args, **kwargs)
  test/integration/test_integration.py:1129: in test_weight_only_quant_force_mixed_mm
      self.assertTrue("mixed_mm" in code)
  E   AssertionError: False is not true

Not sure how this test failure relates to the code change....

[msaroufim]

It's unrelated, we'll merge the fix then your change so no worries!

[msaroufim]

@lancerts can you rebase to main? The error should go away

[lancerts]

@lancerts can you rebase to main? The error should go away

The sync from your end seems still same failure https://github.com/pytorch/ao/actions/runs/9120697036/job/25078538641 So did a rebase on my end. TY