Skip to content

Commit

Permalink
Auto merge of autumnai#47 - autumnai:docs/readme, r=MichaelHirn
Browse files Browse the repository at this point in the history 
docs/readme: add cuDNN v4 to feature matrix

CLOSE: autumnai#43
  • Loading branch information
@homu
homu committed Apr 12, 2016
2 parents 080ffe6 + d6d53b3 commit 6afcb13
Show file tree
Hide file tree
Showing 2 changed files with 37 additions and 36 deletions.
36 changes: 18 additions & 18 deletions README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -24,24 +24,24 @@ This Plugins provides the following operations to the Collenchyma Backend.
Every Operation includes forward + backward. A `-` means not yet implemented.
More information can be found in the [Documentation][docs-ops].

| Operation | CUDA | OpenCL | Native |
|--- |--- |--- |--- |
| Sigmoid | cuDNN v3 | - | Rust |
| SigmoidPointwise | cuDNN v3 | - | |
| ReLU | cuDNN v3 | - | Rust |
| ReLUPointwise | cuDNN v3 | - | |
| Tanh | cudNN v3 | - | Rust |
| TanhPointwise | cudNN v3 | - | |
| | | | |
| Normalization (LRN) | cudNN v3 | - | - |
| | | | |
| Convolution | cudNN v3 | - | - |
| | | | |
| Softmax | cudNN v3 | - | Rust |
| LogSoftmax | cudNN v3 | - | Rust |
| | | | |
| Pooling Max | cudNN v3 | - | - |
| Pooling Avg | cudNN v3 | - | - |
| Operation | CUDA | OpenCL | Native |
|--- |--- |--- |--- |
| Sigmoid | { cuDNN v3, v4 } | - | Rust |
| SigmoidPointwise | { cuDNN v3, v4 } | - | |
| ReLU | { cuDNN v3, v4 } | - | Rust |
| ReLUPointwise | { cuDNN v3, v4 } | - | |
| Tanh | { cuDNN v3, v4 } | - | Rust |
| TanhPointwise | { cuDNN v3, v4 } | - | |
| | | | |
| Normalization (LRN) | { cuDNN v3, v4 } | - | - |
| | | | |
| Convolution | { cuDNN v3, v4 } | - | - |
| | | | |
| Softmax | { cuDNN v3, v4 } | - | Rust |
| LogSoftmax | { cuDNN v3, v4 } | - | Rust |
| | | | |
| Pooling Max | { cuDNN v3, v4 } | - | - |
| Pooling Avg | { cuDNN v3, v4 } | - | - |

Kudos to [ehiggs][ehiggs], for implementing the native Rust operations.

Expand Down
37 changes: 19 additions & 18 deletions src/lib.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -80,24 +80,25 @@
//! This Plugins provides the following operations. (Forward + Backward)
//! A `-` means not yet implemented.
//!
//! | Operation | CUDA | OpenCL | Native |
//! |--- |--- |--- |--- |
//! | Sigmoid | cuDNN v3 | - | Rust |
//! | SigmoidPointwise | cuDNN v3 | - | |
//! | ReLU | cuDNN v3 | - | Rust |
//! | ReLUPointwise | cuDNN v3 | - | |
//! | Tanh | cudNN v3 | - | Rust |
//! | TanhPointwise | cuDNN v3 | - | |
//! | | | | |
//! | Normalization (LRN) | cudNN v3 | - | - |
//! | | | | |
//! | Convolution | cudNN v3 | - | - |
//! | | | | |
//! | Softmax | cudNN v3 | - | Rust |
//! | LogSoftmax | cudNN v3 | - | Rust |
//! | | | | |
//! | Pooling Max | cudNN v3 | - | - |
//! | Pooling Avg | cudNN v3 | - | - |

//! | Operation | CUDA | OpenCL | Native |
//! |--- |--- |--- |--- |
//! | Sigmoid | { cuDNN v3, v4 } | - | Rust |
//! | SigmoidPointwise | { cuDNN v3, v4 } | - | |
//! | ReLU | { cuDNN v3, v4 } | - | Rust |
//! | ReLUPointwise | { cuDNN v3, v4 } | - | |
//! | Tanh | { cuDNN v3, v4 } | - | Rust |
//! | TanhPointwise | { cuDNN v3, v4 } | - | |
//! | | | | |
//! | Normalization (LRN) | { cuDNN v3, v4 } | - | - |
//! | | | | |
//! | Convolution | { cuDNN v3, v4 } | - | - |
//! | | | | |
//! | Softmax | { cuDNN v3, v4 } | - | Rust |
//! | LogSoftmax | { cuDNN v3, v4 } | - | Rust |
//! | | | | |
//! | Pooling Max | { cuDNN v3, v4 } | - | - |
//! | Pooling Avg | { cuDNN v3, v4 } | - | - |
//!
//! [collenchyma]: https://github.com/autumnai/collenchyma
//! [collenchyma-docs]: http://autumnai.github.io/collenchyma
Expand Down

0 comments on commit 6afcb13

Please sign in to comment.