Use exp2 for mx scaling

[drisspg]

Stacked PRs:

• ->Use exp2 for mx scaling #1530

---

Use exp2 for mx scaling

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 2 New Failures

As of commit 9d068d5 with merge base 11333ba ():

NEW FAILURES - The following jobs have failed:

• Run Regression Tests / test (CUDA 2.4, linux.g5.12xlarge.nvidia.gpu, torch==2.4.0, cuda, 12.1) / linux-job (gh)
  RuntimeError: Command docker exec -t 25160cd995866d94e61f9217d8e188a2521250ee5f5b440a088e4cb5f52aa4e7 /exec failed with exit code 1
• Run Regression Tests / test (CUDA 2.5.1, linux.g5.12xlarge.nvidia.gpu, torch==2.5.1 --index-url https://download.pytorch... / linux-job (gh)
  RuntimeError: Cannot access data pointer of Tensor (e.g. FakeTensor, FunctionalTensor). If you're using torch.compile/export/fx, it is likely that we are erroneously tracing into a custom kernel. To fix this, please wrap the custom kernel into an opaque custom op. Please see the following for details: https://pytorch.org/tutorials/advanced/custom_ops_landing_page.html

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

[vkuzo]

TIL about torch.exp2

[vkuzo]

should ensure CI is green though