https://github.com/torch/cunn
Revision bedeebfaa27ac91f2ebcf30e21fbe243270e2bb8 authored by Gregory Chanan on 26 October 2016, 21:42:28 UTC, committed by Gregory Chanan on 08 November 2016, 21:01:06 UTC
We don't support SparseLInear with fp16 because of lack of cusparseHcsrmm (or equivalent Ex function) until CUDA 8.0.
1 parent f43f9aa
Tip revision: bedeebfaa27ac91f2ebcf30e21fbe243270e2bb8 authored by Gregory Chanan on 26 October 2016, 21:42:28 UTC
Add generic support for SparseLinear.
Add generic support for SparseLinear.
Tip revision: bedeebf
File | Mode | Size |
---|---|---|
doc | ||
lib | ||
rocks | ||
.gitignore | -rw-r--r-- | 19 bytes |
CMakeLists.txt | -rw-r--r-- | 850 bytes |
CONTRIBUTING.md | -rw-r--r-- | 4.9 KB |
DataParallelTable.lua | -rw-r--r-- | 21.3 KB |
LICENSE | -rw-r--r-- | 1.9 KB |
README.md | -rw-r--r-- | 3.1 KB |
THCUNN.lua | -rw-r--r-- | 5.3 KB |
init.lua | -rw-r--r-- | 185 bytes |
test.lua | -rw-r--r-- | 206.4 KB |
test_DataParallelTable.lua | -rw-r--r-- | 23.1 KB |
Computing file changes ...