add _c10d_functional._wrap_tensor_autograd impl for FP8 to fix ci

[liangel-02]

as title

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 9 New Failures, 4 Pending

As of commit ce19814 with merge base 886ac8e ():

NEW FAILURES - The following jobs have failed:

• Run 4xH100 tests / test (H100, linux.aws.h100.4, --pre torch torchvision torchaudio --index-url https://download.pyt... / linux-job (gh)
  RuntimeError: Command docker exec -t 939fd06c5854d93ab6967c6e794695ee66490157246626eb5309e6abcda7bb73 /exec failed with exit code 1
• Run Regression Tests / test (CPU 2.6, linux.4xlarge, torch==2.6.0 --index-url https://download.pytorch.org/whl/cpu, cpu) / linux-job (gh)
  RuntimeError: Command docker exec -t 287a1c709140e37b6a0ce09b38dd0e57b1bf5a712571adde36450bedfac9e77c /exec failed with exit code 2
• Run Regression Tests / test (CPU 2.7, linux.4xlarge, torch==2.7.1 --index-url https://download.pytorch.org/whl/cpu, cpu) / linux-job (gh)
  RuntimeError: Command docker exec -t e94a87cea139081d1e9967c4d590bd5196e68e788ddb432defc75f0afb9708ef /exec failed with exit code 2
• Run Regression Tests / test (CPU 2.8, linux.4xlarge, torch==2.8.0 --index-url https://download.pytorch.org/whl/cpu, cpu) / linux-job (gh)
  RuntimeError: Command docker exec -t b785a0a667256fccf23a999e08ad1c3885b0c1416ce8d4b7e236899dd536d7b7 /exec failed with exit code 2
• Run Regression Tests / test (CPU 2.9, linux.4xlarge, torch==2.9.1 --index-url https://download.pytorch.org/whl/cpu, cpu) / linux-job (gh)
  RuntimeError: Command docker exec -t 743c8549766625af84263f925c3f5439b0836292a08961a8a491d66e32ceabc8 /exec failed with exit code 2
• Run Regression Tests / test (CUDA 2.6, linux.g5.12xlarge.nvidia.gpu, torch==2.6.0, cuda, 12.6) / linux-job (gh)
  RuntimeError: Command docker exec -t c8aafaf4422e5193da539418eede2b707e1d3663851c281b23d1f8ed863edad9 /exec failed with exit code 2
• Run Regression Tests / test (CUDA 2.7, linux.g5.12xlarge.nvidia.gpu, torch==2.7.1, cuda, 12.6) / linux-job (gh)
  RuntimeError: Command docker exec -t 07deedea81d638d75324cfa3a6c5c2c455b1df0f512606b2646b9ac562238182 /exec failed with exit code 2
• Run Regression Tests / test (CUDA 2.8, linux.g5.12xlarge.nvidia.gpu, torch==2.8.0, cuda, 12.6) / linux-job (gh)
  RuntimeError: Command docker exec -t 01ece6ff04986df30b5cc6e6d6d6fd410ae04082815d56be721714bdbd8afb40 /exec failed with exit code 2
• Run Regression Tests / test (CUDA 2.9, linux.g5.12xlarge.nvidia.gpu, torch==2.9.1, cuda, 12.6) / linux-job (gh)
  RuntimeError: Command docker exec -t edbdcc0197c4a29aed26d5649d6aabc2963bc142890a27a86cb3e3c031769d1f /exec failed with exit code 2

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

[vkuzo]

#3548 does the same thing

[liangel-02]

@vkuzo ok great closing this then, thanks!