Dispatcher Integration
PrivateUse1
Backend menggunakan slot backend:
cpp
PrivateUse1yang disediakan oleh PyTorch.
Registrasi Backend
python
rename_privateuse1_backend("opencl")Dispatch Flow
mermaid
flowchart LR
A[Tensor Operation] --> B[Dispatcher]
B --> C{Device Type}
C -->|CPU| D[CPU Kernel]
C -->|CUDA| E[CUDA Kernel]
C -->|OpenCL| F[OpenCL Kernel]Operator Registration
Operator didaftarkan menggunakan:
TORCH_LIBRARYTORCH_LIBRARY_IMPL
Device Guard
Backend juga memerlukan:
- device guard
- stream handling
- synchronization primitive