System Overview

High Level Architecture

flowchart TD
    A[Python API] --> B[torch_opencl]
    B --> C[PyTorch Dispatcher]
    C --> D[PrivateUse1 Backend]
    D --> E[Runtime Layer]
    D --> F[Allocator]
    D --> G[Kernel Registry]
    E --> H[OpenCL Driver]
    F --> H
    G --> H

Tujuan Backend

Project ini menyediakan backend OpenCL agar PyTorch dapat menjalankan operasi tensor di GPU/device yang mendukung OpenCL tanpa bergantung pada CUDA.

Komponen Utama

Python Layer

Layer Python bertanggung jawab untuk:

• inisialisasi extension
• registrasi backend
• expose API helper

Dispatcher Layer

PyTorch dispatcher menentukan implementasi operator berdasarkan device type tensor.

Runtime Layer

Runtime mengelola:

• OpenCL platform
• device enumeration
• command queue
• synchronization

Allocator Layer

Allocator bertugas menerjemahkan memory model PyTorch ke cl::Buffer.

Struktur Folder

csrc/
├── aten/
├── runtime/
├── kernels/
└── _C.cpp

torch_opencl/
├── __init__.py
└── opencl/