test fsdp2 Moe

[weifengpy]

No description provided.

[pytorch-bot]

🔗 Helpful Links

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

• 📄 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, 1 Cancelled Job, 1 Unrelated Failure

As of commit ca2ca62 with merge base abffabb ():

NEW FAILURES - The following jobs have failed:

• Code Analysis with Ruff / build (3.9) (gh)
  Process completed with exit code 1.
• Run TorchAO Experimental Tests / test-cpu-ops (linux.arm64.2xlarge) (gh)
  /home/ec2-user/actions-runner/_work/ao/ao/torchao/experimental/../../torchao/experimental/ops/embedding_lut/op_embedding_groupwise_lowbit_lut-impl.h:227:50: error: ‘packed_embedding_size’ is not a member of ‘torchao::kernels::cpu::aarch64::embedding’

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

• Run TorchAO Experimental Tests / test-cpu-ops (macos-14) (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 1xH100 Tests / test (H100, linux.aws.h100, --pre torch torchvision torchaudio fbgemm-gpu-genai --index-url https... / linux-job (gh) (trunk failure)
  test/dtypes/test_affine_quantized_float.py::TestAffineQuantizedFloat8Compile::test_expected_kernels_on_gpu_granularity1_float8_config_version_2

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

[weifengpy]

trying to repro #2673 but this passed. need to sync more with @danielvegamyhre