Change TORCH_LIBRARY to TORCH_LIBRARY_FRAGMENT

[metascroy]

This is to prevent conflicts between CPU/MPS because torch requires all ops be defined in same block.

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 3 New Failures

As of commit d791cb9 with merge base c8eb8d3 ():

NEW FAILURES - The following jobs have failed:

• Code Analysis with Ruff / build (3.9) (gh)
  Process completed with exit code 1.
• PR Label Check / Check PR Labels (gh)
  Process completed with exit code 1.
• Run TorchAO Experimental Tests / test (macos-14) (gh)
  Process completed with exit code 139.

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

[metascroy]

@manuelcandales how do I test the change? Is there a script I should run?

[manuelcandales]

cd ao/torchao/experimental/ops/mps
bash build.sh
python test/test_lowbit.py