search

Idein / py-videocore6

Python library for GPGPU programming on Raspberry Pi 4
https://idein.jp
GNU General Public License v2.0
247 stars 28 forks source link

possible usage in pytroch? #55

Open Hamidreza-Ramezani opened 2 years ago

Hamidreza-Ramezani commented 2 years ago

Hello,

Thank you for creating this great library. I was wondering if it can be used in torch as well for machine learning inference task (on Raspberry pi4)? If so, do you have any example for that?

Cheers,

notogawa commented 2 years ago

This library is just a VideoCore VI QPU assembler. To use VideoCore VI GPGPU in torch (or other frameworks), you must implement DNN operations and replace torch's one. Please see https://github.com/Idein/qmkl6 as such an operation (GEMM) implementation example.