fix nf4 test that is failing in CI

[vkuzo]

Summary:

Fixes

python test/dtypes/test_nf4.py TestComm.test_comm

This test is broken in CI for PyTorch 2.9, fixing.

AI assisted, logs:
https://gist.github.com/vkuzo/6526e8e14cd7797ce6c284d5bfb57250

Test Plan:

python test/dtypes/test_nf4.py

Reviewers:

Subscribers:

Tasks:

Tags:

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❗ 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

• AWS was down, GHA infrastructure effected / recovering

❌ 1 Cancelled Job, 2 Pending, 1 Unrelated Failure

As of commit ccf420e with merge base 158e72c ():

CANCELLED JOB - The following job was cancelled. Please retry:

• Run Regression Tests (aarch64) / test-cpu-ops (linux.arm64.2xlarge) (gh)
  ##[error]The operation was canceled.

BROKEN TRUNK - The following job failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

• Run Regression Tests (aarch64) / test-cpu-ops (macos-14) (gh) (trunk failure)
  /Users/runner/miniconda3/envs/venv/lib/python3.10/site-packages/executorch/share/cmake/../../include/executorch/runtime/core/memory_allocator.h:143:21: error: no matching function for call to 'mul_overflows'

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