- Notifications
You must be signed in to change notification settings - Fork0
Autograd and GPGPU library for dynamic neural networks in D
NotificationsYou must be signed in to change notification settings
ShigekiKarita/grain2
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
Autograd and GPGPU library for dynamic neural networks in D.This project is a successor ofgrain.
- @nogc support
- heterogeneous device (CPU/CUDA/OpenCL) support
- multiple CUDA device support
- see also document
- dpp master
- CPU backend
- BLAS library (e.g., OpenBLAS, MKL)
- CUDA backend
- CUDA header/library v10-
- CUDNN header/library v7-
- OpenCL backend
- OpenCL header/library v1.2
- test
dub test grain2:coreanddub test grain2:cuda - generate ddoc
make doc
- heterogeneous multi-dimentional structure (grain.tensor.Tensor)
- reference count-based memory management
- multiple CUDA devices support
- core autograd (forward/backward chain)
- dataset API (MNIST, PTB, etc)
- more memory manegement policy (garbage collection, memory pool)
- JIT compilation of computational graph
- OpenCL/HIP support
- Q. My local CUDA/OpenCL libraries are different from files under
grain2/dpp/build- A. Installdpp (commit 821a5d4) and regenerate them by
$ cd dpp; make -j cuda (or opencl)
- A. Installdpp (commit 821a5d4) and regenerate them by