Duplicate pt2e quant code from pytorch/pytorch to pytorch/ao

[jerryzh168]

Summary:
First step of https://dev-discuss.pytorch.org/t/torch-ao-quantization-migration-plan/2810

core logic of pt2e are duplicated, and also ported test here

Next:

• move meta internal callsites to depend on torchao
• Docs will be migrated separately

Test Plan:
pytest test/quantization/pt2e_flow

Reviewers:

Subscribers:

Tasks:

Tags:

[pytorch-bot]

🔗 Helpful Links

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

• 📄 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

As of commit 297c990 with merge base f38c272 ():

NEW FAILURES - The following jobs have failed:

• Run Regression Tests / test (CPU 2.3, linux.4xlarge, torch==2.3.0 --index-url https://download.pytorch.org/whl/cpu, cpu) / linux-job (gh)
  RuntimeError: Command docker exec -t eb19990bdcd7202113a1fb521b238076901e938c863107b7129ede8eabf86ef2 /exec failed with exit code 2
• Run Regression Tests / test (CPU 2.4, linux.4xlarge, torch==2.4.0 --index-url https://download.pytorch.org/whl/cpu, cpu) / linux-job (gh)
  RuntimeError: Command docker exec -t 8347702aefd658c578531390c513f70af2b7ddc2711ec6a3f412a0980ce585d4 /exec failed with exit code 2
• Run Regression Tests / test (CPU 2.5.1, linux.4xlarge, torch==2.5.1 --index-url https://download.pytorch.org/whl/cpu, cpu) / linux-job (gh)
  RuntimeError: Command docker exec -t 0368994e1955293839101b6c443fdc077c9509d04be653b49da73cd437392fba /exec failed with exit code 2
• Run Regression Tests / test (CUDA 2.3, linux.g5.12xlarge.nvidia.gpu, torch==2.3.0, cuda, 12.1) / linux-job (gh)
  RuntimeError: Command docker exec -t 799ade81f97ad3694b553310952ea3fd35b2880ad8b029294db43af5c2f5e218 /exec failed with exit code 2
• 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 a8c4f604ae2daeadff9b202a1962b1dd8fb812f6630e0dfafe5a701834edffb1 /exec failed with exit code 2
• 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: Command docker exec -t fc85720b69653862321570ddac2d142adbf60bb74da755b58de025235dcc04ac /exec failed with exit code 2
• Run Regression Tests / test-nightly (CPU Nightly, linux.4xlarge, --pre torch --index-url https://download.pytorch.org/wh... / linux-job (gh)
  RuntimeError: Command docker exec -t ce0e941d202756b85b0299b03fde81146a3b42e6faddc0455bde39a991cbf53e /exec failed with exit code 2
• Run Regression Tests / test-nightly (CUDA Nightly, linux.g5.12xlarge.nvidia.gpu, --pre torch --index-url https://downloa... / linux-job (gh)
  RuntimeError: Command docker exec -t c1bcfc2334cca73bea6dc4795050f5c257e4faf459dca87f5feedb94d45ae0f2 /exec failed with exit code 2
• Run TorchAO Experimental Tests / test-cpu-ops (macos-14) (gh)
  torchao/experimental/tests/test_int8_dynamic_activation_intx_weight.py::TestInt8DynamicActivationIntxWeight::test_export_QDQLayout

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