An OpenCL backend for torch.
It's a high-performance matrix library for OpenCL, that runs on your GPU(s) harnessing the massive computational capacity that they provide.
Most of the standard operations in torch are supported. If there are any missing that you need, please raise an issue.
Most things really :-) Detailed description at ImplementedDetails.md. Please don't hesitate to raise an issue for anything that's missing that you would like to see added.
IMPORTANT! THIS HAS CHANGED. Please install a specific Torch distro, as described below. Simply doing luarocks install cltorch
is no longer supported
Please see distro-cl for installation instructions.
The following features are either cltorch-specific, or do not exist in cutorch:
feature | in torch? | in cutorch? | in cltorch? |
---|---|---|---|
apply/map/map2 | Yes | Via an extension | Yes |
profiling tools | Via CUDA nvvp |
Yes | |
point tensors | Yes | ||
custom user kernels | Not applicable | Via cutorch-rtc | Yes |
apply
, map
, map2
exist in torch, but how to make them work on the GPU? Cannot just pass in lua functions.
What we do is, you can provide opencl code directly to apply_on_gpu, map_on_gpu and map2_on_gpu, as a string expression. This will run on the gpu, at full speed. Examples, for x
, y
, z
being identically sized torch.ClTensor
s:
x:apply_on_gpu("x = sqrt(x + 3.5)")
x:map_on_gpu(y, "x = 1000 * x + y * 10")
x:map2_on_gpu(y, z, "x = sqrt(1000 * x + y * 10 + z * z)")
- note that the variables in the OpenCL string expression must be named as above, ie
x
,y
,z
. For convenience, these were named the same as the tensors in the example. If the tensors have different names, please continue to usex
,y
,z
in the expressions, eg:
a:apply_on_gpu("x = sqrt(x + 3.5)")
a:map_on_gpu(b, "x = 1000 * x + y * 10")
a:map2_on_gpu(b, c, "x = sqrt(1000 * x + y * 10 + z * z)")
Following tools are available to aid with profiling:
Method | Description |
---|---|
cltorch.setProfiling(1) |
turn on opencl kernel profiling |
cltorch.dumpProfiling() |
dump opencl kernel profiling timings since last call |
cltorch.setEnableTiming(1) |
enable collection of cumulative wall-clock timings for cltorch code |
cltorch.dumpTimings() |
dump cumulative wall-clock timings for cltorch code |
cltorch.setTrace(1) |
print all gpu buffer allocations and copies between host/gpu |
OpenCL natively provides facilities to measure the execution time of kernels, without needing to call cltorch.finish()
or similar first, using clGetEventProfilingInfo. In cltorch, you dont need to know how this works ;-) Simply call, at the start of your code:
cltorch.setProfiling(1)
Then, after running the piece of code under scrutiny, simply call:
cltorch.dumpProfiling()
Timings are cumulative across multiple calls to the same kernel.
This uses the wall-clock times to measure the elapsed time in different sections of cltorch code. The way it works is, each time the cltorch c++ code calls StatefulTimer::instance()->timeCheck("some status")
, the wall-clock time since the last call to ->timeCheck()
will be added to the cumulative time for some status
. You can pass any status as a string. Then, after running the piece of code under the scrutiny, in your Lua program, simply call cltorch.dumpTimings()
to dump these cumulative timings.
Update: please first call cltorch.setEnableTiming(true)
to enable collection of timing information. This is global across all devices.
You can log all GPU buffer allocations, copies to host, and copies to GPU device. Simply call:
cltorch.setTrace(1)
Any buffer allocations, and copies between host and device, will now be printed to stdout.
Point tensors help to eliminate pipeline stalls associated with ReduceAll operations such as sometensor:sum()
. Why does :sum()
cause pipeline stalls, and how do point tensors eliminate this source of stalls?
If we send a single instruction (a kernel) to the gpu, there will be some latency whilst the instruction arrives at the gpu, and starts running, and some more latency after the calculations have finished, whilst the results are retrieved back from the GPU. Maybe we send:
a:add(1)
We can draw a picture of what happens. Time is towards the right. GPU is at the top. CPU at the bottom:
But we can send lots of instructions, without waiting for the earlier ones to finish. Maybe we do:
a:add(b)
a:mul(3)
b:mul(a)
c:add(a)
This might look like this, we dont have to wait for the previous instruction to finish:
But now imagine what happens if we process the following instruction:
a:div(a:sum())
- a:sum() is going to take the sum of all the elements in a
- a:div(a:sum()) is then going to divide all the elements of a by this sum
- it looks innocent enough
- but notice that we cannot send the
a:div
instruction until thea:sum()
results have come back - so we have to wait for
a:sum()
to finish processing, and for the results to come back, before we can continue
Classic reduceall => Massive pipeline stall
Point tensors eliminate this. When we do the reduceall, the :sum()
operation, we keep the results on the gpu, like this:
c = torch.Tensor(20,30):uniform():cl() -- create a tensor on the GPU
res = torch.ClTensor() -- create a point tensor on the GPU
res:sum(c) -- sum c, and keep the result in res, on the GPU
res
is a point tensor. It has zero dimensions. It contains a single scalar float. It stays on the GPU. We can feed it into other operations as follows:
c:div(res) -- divide c by res
We can send this instruction straight away, even before the first :sum(c)
instruction has arrived at the GPU. So, no more stall.
By the way, it's possible to print the value of a point tensor, by printing it, or calling the :s()
operator. Normally you wouldnt do this except during debugging though, since obviously this will need to wait for the gpu operation to finish, and for the data to come all the way back from the GPU :-)
Custom user kernels let you run OpenCL code directly from Torch Lua! Of course, you can already do this with apply
, map
, and map2
, see above. But now you can provide whole kernel functions, and other functions, and pass ClTensors into these kernels!
Example of how to use:
require 'cltorch'
k = torch.ClKernel({input={nElements='int', input='torch.ClTensor'},output={output='torch.ClTensor'},src=[[
int linearId = get_global_id(0);
if(linearId < nElements) {
output_data[linearId] = input_data[linearId] + 3.0f;
}
]]})
print('k', k)
k:print()
x = torch.ClTensor({3,5,2})
y = torch.ClTensor({6,4,2})
print('x before\n', x)
print('y before\n', y)
k:run({nElements=3, input=x, output=y})
print('y after\n', y)
Output from this example:
Using Intel platform: Intel Gen OCL Driver
Using device: Intel(R) HD Graphics BroadWell U-Processor GT2
k torch.ClKernel
Original source
===============
int linearId = get_global_id(0);
if(linearId < nElements) {
output_data[linearId] = input_data[linearId] + 3.0f;
}
Generated source
================
typedef struct THClTensorInfoCl {
unsigned int sizes[25];
unsigned int strides[25];
int offset;
int dims;
} TensorInfoCl;
kernel void user_kernel(
global struct THClTensorInfoCl *input_info, global float * input_data,
int nElements,
global struct THClTensorInfoCl *output_info, global float * output_data
) {
int linearId = get_global_id(0);
if(linearId < nElements) {
output_data[linearId] = input_data[linearId] + 3.0f;
}
}
x before
3
5
2
[torch.ClTensor of size 3]
y before
6
4
2
[torch.ClTensor of size 3]
y after
6
8
5
[torch.ClTensor of size 3]
If you want, you can specify the number of workgroups, and the workgroupsize, yourself:
k:run({nElements=3, input=x, output=y}, {numWorkgroups=10, workgroupSize=32}
- It is possible to load cutorch and cltorch at the same time, if you wish
- If you do this, please load cutorch first, and then load cltorch second
- If you get errors about #1 argument to copy should be tensor, but is userdata, then please double-check that cutorch is
required
d before cltorch (they each monkey-patch torch, but since cutorch was written first, it assumes there is no monkey-patch conflict)
cltorch uses the following libraries. These are automatically built as part of cltorch build process:
- clBLAS - provides GPU-based matrix operations, such as multiplication
- EasyCL - provides an abstraction layer over the low-level OpenCL API
- clew - similar to glew, means that cltorch can be loaded without any OpenCL library/runtime being present
At runtime, if you want to call any of the cltorch methods, you will also need:
- OpenCL-compatible GPU
- OpenCL library/driver (normally provided by the GPU vendor)
You might or might not find ContributorGuidelines.md useful. Not required reading, but it is there if you want to see my own thoughts and ideas on how I am currently approaching cltorch development, and cutorch-porting.
Also, some more technical guidelines on porting, in the clnn repository, at porting-guidelines.md.
There is an OpenCL backend for nn
and nngraph
at clnn.
There is an HCC backend for Torch at: hctorch
- 20 August 2016:
- tests updated to use TestSuite.lua
- renamed
apply
,map
,map2
toapply_on_gpu
,map_on_gpu
,map2_on_gpu
, and added theapply
method from cutorch that first copies to main memory, then runs on the cpu, then copies back again, for compatibility with cutorch
- 31 April 2016:
- Re-applied:
- 27 March 2016:
- migrated from clBLAS 2.4 to clBLAS 2.11/develop. This migration is not set in stone, depends on how well that works. However, there is a bug in 2.4 for certain configurations of matrix multiplication, and its not obvious how to fix that, so maybe using 2.11/develop is the easiest way forward?
- 27 March 2016:
- Re-applied:
- 30 April 2016:
- rolled back to as of 3 March 2016, to use specific torch release, so it doesnt keep changing whilst I'm at work :-)
- 3 March 2016:
- runs on Mac OS X, without needing
LD_LIBRARY_PATH
, ie RPATH works now. Hopefully :-)
- runs on Mac OS X, without needing
- 3rd January, 2016:
- created Mac build on Travis, https://travis-ci.org/hughperkins/cltorch , which passes (at time of writing)
- 27th December:
- added FFI functions
:data()
and:cdata()
, which means that Element Research's rnn now works withclnn
- added FFI functions
- 23rd October:
- removed
:csub()
and:neg()
from the "cltorch-specific features" section, since integrated into torch now :-) pull request 392
- removed
- 3rd October:
- Added
:mean()
and:mean(d)
- Added
:atan2(x,y)
- Added
x:sign()
andtorch.sign(x)
- Added
norm(...)
- Added
- 20th September:
- Ported fix to
addcdiv
andaddcmul
reshape from cutorch commit 59a1cb05 - Added ClStorage:__index() and ClTensor:__index()
- Added ClStorage:__newindex() and ClTensor:__newindex()
- Ported fix to
- 19th September:
- Added guards around many functions, so that c++ exceptions are converted to torch errors now, and display something more meaningful than just 'c++ exception' :-P
- Please feel free to raise an issue for any exceptions which are not guarded yet
- Added guards around many functions, so that c++ exceptions are converted to torch errors now, and display something more meaningful than just 'c++ exception' :-P