OpenCL out of tree backend of PyTorch
pytorch-edge is tested on following devices:
| vendor | device |
|---|---|
| Nvidia | GTX 960, A100/A5000, RTX 3060 |
| AMD | rx 6600XT/rx560 |
| Intel | HD 530/8570, UHD 630/770, Arc A380 |
| Glenfly | Arise 2030 |
| Innosilicon | Fantasy II-M |
| ZhaoXin | ZX C-1190 |
git clone --recursive https://github.com/JingliangGao/pytorch-edge.git
cd pytorch-edge/
chmod +x build-for-debug.sh && ./build-for-debug.sh- Python example
cd examples/python/
python3 tensor_add.py