Abstract GPU Array package for Julia's various GPU backends.
Julia offers great advantages for programming the GPU. This blog post outlines a few of those.
E.g., we can use Julia's JIT to generate optimized kernels for map/broadcast operations.
This works even for things like complex arithmetic, since we can compile what's already in Julia Base. This isn't restricted to Julia Base, GPUArrays works with all kind of user defined types and functions!
GPUArrays relies heavily on Julia's dot broadcasting. The great thing about dot broadcasting in Julia is, that it actually fuses operations syntactically, which is vital for performance on the GPU. E.g.:
out .= a .+ b ./ c .+ 1 #turns into this one broadcast (map): broadcast!(out, a, b, c) do a, b, c a + b / c + 1 end
Will result in one GPU kernel call to a function that combines the operations without any extra allocations. This allows GPUArrays to offer a lot of functionality with minimal code.
Also, when compiling Julia for the GPU, we can use all the cool features from Julia, e.g. higher order functions, multiple dispatch, meta programming and generated functions. Checkout the examples, to see how this can be used to emit specialized code while not losing flexibility:
In theory, we could go as far as inspecting user defined callbacks (we can get the complete AST), count operations and estimate register usage and use those numbers to optimize our kernels!
Interface offered for all backends:
map(f, ::GPUArray...) map!(f, dest::GPUArray, ::GPUArray...) broadcast(f, ::GPUArray...) broadcast!(f, dest::GPUArray, ::GPUArray...) mapreduce(f, op, ::GPUArray...) # so support for sum/mean/minimum etc comes for free getindex, setindex!, push!, append!, splice!, append!, copy!, reinterpret, convert From (CL/CU)FFT fft!/fft/ifft/ifft! and the matching plan_fft functions. From (CL/CU)BLAS gemm!, scal!, gemv! and the high level functions that are implemented with these, like A * B, A_mul_B!, etc.
Working with immutable isbits (not containing pointers) type should be completely supported
with non-allocating code (so no constructs like
x = [1, 2, 3]). Note that tuples are isbits, so this works x = (1, 2, 3).
Transpiler/OpenCL has problems with putting GPU arrays on the gpu into a struct - so no views and actually no multidimensional indexing. For that
size is needed which would need to be part of the array struct. A fix for that is in sight, though.
JLArray is a
GPUArray which doesn't run on the GPU and rather uses Julia's async constructs as its backend. It serves as a fallback for testing compatibility with
GPUArrays in cases where a GPU does not exist and as a reference implementation. It is constructed as follows:
gA = JLArray(A)
See CuArrays or CLArrays for installation instructions.
4 days ago