Tensor parallel support for uintx dtype

[jainapurva]

No description provided.

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 4 New Failures, 1 Unrelated Failure

As of commit 41f4d9d with merge base 6ffe236 ():

NEW FAILURES - The following jobs have failed:

• Run Regression Tests / test (CUDA 2.3, linux.g5.12xlarge.nvidia.gpu, torch==2.3.0, cuda, 12.1) / linux-job (gh)
  test/dtypes/test_affine_quantized_tensor_parallel.py::TestUintxAffineQuantizedTensorParallel::test_tp_bfloat16
• Run Regression Tests / test (CUDA 2.4, linux.g5.12xlarge.nvidia.gpu, torch==2.4.0, cuda, 12.1) / linux-job (gh)
  test/dtypes/test_affine_quantized_tensor_parallel.py::TestUintxAffineQuantizedTensorParallel::test_tp_bfloat16
• 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)
  test/dtypes/test_affine_quantized_tensor_parallel.py::TestUintxAffineQuantizedTensorParallel::test_tp_bfloat16
• Run Regression Tests / test-nightly (CUDA Nightly, linux.g5.12xlarge.nvidia.gpu, --pre torch==2.7.0.dev20250122 --index-... / linux-job (gh)
  test/dtypes/test_affine_quantized_tensor_parallel.py::TestUintxAffineQuantizedTensorParallel::test_tp_bfloat16

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

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

• Code Analysis with Ruff / build (3.9) (gh) (trunk failure)
  Process completed with exit code 1.

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