Skip to content
forked from bluekidds/jtorch

Jonathan's Torch7 Utility Library (C++, OpenCL)

Notifications You must be signed in to change notification settings

diz-vara/jtorch

 
 

Repository files navigation

jtorch - Jonathan's Torch7 Utility Library (C++, OpenCL)


Overview

Torch7 (http://www.torch.ch/) is an AMAZING machine learning library written and maintained by some very smart people :-) For me it's only downside is that interfacing with it from C++ on Windows 7 (and other operating systems other than Mac OS X and Linux) is difficult (if not sometimes impossible). This library is a C++ framework for doing the forward propagation of various torch modules. It uses OpenCL to perform the forward prop on the GPU. I have even found that some of the OpenCL modules here are actually faster than the Torch7 CUDA modules on Linux. With this said, you should profile torch vs jtorch and make sure there are no super slow modules in this library (since I haven't spent all that much time optimizing GPU code).

Please note that this is not supposed to be a replacement for Torch7. There is no back propagation (so no learning), and only a very limited subset of the modules are implemented. Additionally, none of the modules support batches (so single sample FPROP only). The use-case for this library is for people who do model development on Linux, but want to run real-time FPROP of their models on other operating systems.

The library consists of a simple lua codebase for recursively saving a torch model to a compact binary format (all in the ./lua folder):

-- TORCH USAGE:
dofile("./lua/save_nn_node.lua")
model = nn.Sequential()
model.add(x)  -- Add all the model components here
jtorch_root = "path_to/jtorch/"
dofile("../jtorch/jtorch.lua")
saveModel(model, "my_model.bin")

The library also contains a CPP framework for loading it and doing the forward prop. See jtorch_test for more details of usage. It uses OpenCL for all GPU computing. The following stages have full implementations (without batch support):

  • CAddTable
  • Identity
  • Linear
  • Parallel
  • Reshape
  • SelectTable
  • Sequential
  • SpatialContrastiveNormalization
  • SpatialConvolution
  • SpatialConvolutionMM --> (using clBLAS)
  • SpatialConvolutionMap --> (only on CPU)
  • SpatialDivisiveNormalization
  • SpatialLPPooling --> (only on CPU)
  • SpatialMaxPooling
  • SpatialSubtractiveNormalization
  • SpatialUpSamplingNearest
  • Tanh
  • Threshold

The following stages have partial implementations:

  • JoinTable --> Only joins along the top most dimension are allowed for now
  • Transpose --> Just a pass through stage. Again, it just points to the previous stage's output.

Compilation

Building jtorch uses Visual Studio 2012 on Windows, and cmake + gcc 4.7 (or greater) on Mac OS X. The only dependencies are the jcl library (which should compile out of the box) and clBLAS. See http://github.com/jonathantompson/jcl and https://github.com/clMathLibraries/clBLAS for more details (also for clBLAS I have a "Compiling_clBLAS.txt" helper if you get stuck).

VS2012 and cmake expect a specific directory structure:

  • \jcl\
  • \jtorch\
  • \clBLAS\

So jtorch, clBLAS and jcl must exist at the same directory level.

Windows:

  • Follow all the compilation steps in https://github.com/jonathantompson/jcl/blob/master/README.md
  • Follow all the compilation steps in Compiling_clBLAS.txt
  • Compiling the library:
    • open jtorch.sln
    • right click "jtorch_test" in the Solution Explorer -> "Set as StartUp Project"
    • From the file menu: DEBUG -> Start Debugging... (or press F5)
    • The tests should run and at the end of each test there should be a "PASSED" printout (otherwise it will print FAILED).

MAC OS X:

  • Before integration of clBLAS everything compiled and ran correctly. However, post clBLAS I have not tested it on Mac OS X or Linux.

Style

This project follows the Google C++ style conventions:

http://google-styleguide.googlecode.com/svn/trunk/cppguide.xml

License

This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.

This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License along with this program. If not, see http://www.gnu.org/licenses/.

About

Jonathan's Torch7 Utility Library (C++, OpenCL)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C++ 77.9%
  • Lua 11.1%
  • C 11.0%