[ET Device Support] Define AOT device copy ops registry

[Gasoonjia]

Stack from ghstack (oldest at bottom):

• [ET Device Support] Add device tensor helper functions to TensorPtr API #18761
• [ET Device Support] make device support config method-based #19582
• [ET Device Support] Add ExecutorchBackendConfig flags for skipping H2D/D2H copies #18760
• [ET Device Support] PropagateDevicePass inserts H2D/D2H copy ops at delegate boundaries #18730
• [ET Device Support] Define et_copy runtime h2d and d2h copy ops #18729
• -> [ET Device Support] Define AOT device copy ops registry #18728
• [ET Device Support] CudaAllocator: device memory allocator for CUDA backend #18477
• [ET Device Support] Module: allocate device memory for planned buffers #18476

Define et_copy._h2d_copy and et_copy._d2h_copy custom ops for explicit
host-to-device and device-to-host data transfer at delegate boundaries.

Follows dim_order_ops_registry.py pattern:

• Defines functional and out variants for both ops
• Tracing implementations: clone/copy (CPU-only during tracing)
• Registered via torch.library

Differential Revision: D99636779

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/18728

• 📄 Preview Python docs built from this PR

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

❌ 1 New Failure, 40 Pending, 4 Unrelated Failures, 1 Unclassified Failure

As of commit fd5b5c3 with merge base a83e7c4 ():

NEW FAILURE - The following job has failed:

• pull / test-qnn-python-imports-linux / linux-job (gh)
  RuntimeError: Command docker exec -t b551c13eeda52bc1ca70d44647972c3943ed41f76dfe16bd0aa8cf172df8dec7 /exec failed with exit code 1

UNCLASSIFIED FAILURE - DrCI could not classify the following job because the workflow did not run on the merge base. The failure may be pre-existing on trunk or introduced by this PR:

• Check Labels / Check labels (gh) (this job did not run on the merge base, so DrCI cannot tell whether the failure is pre-existing)

FLAKY - The following job failed but was likely due to flakiness present on trunk:

• trunk / unittest-release / macos / macos-job (gh) (similar failure)
  ##[error]The operation was canceled.

BROKEN TRUNK - The following jobs failed but was present on the merge base:

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

• pull / test-parakeet-xnnpack-linux / linux-job (gh) (trunk failure)
• pull / unittest / macos / macos-job (gh) (trunk failure)
  ##[error]The operation was canceled.
• pull / unittest-editable / macos / macos-job (gh) (trunk failure)
  ##[error]The operation was canceled.

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

[github-actions]

This PR needs a release notes: label

If your change should be included in the release notes (i.e. would users of this library care about this change?), please use a label starting with release notes:. This helps us keep track and include your important work in the next release notes.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "release notes: none"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.

[digantdesai]

Review automatically exported from Phabricator review in Meta.

[JacobSzwejbka]

Review automatically exported from Phabricator review in Meta.