CUBLAS.jl icon indicating copy to clipboard operation
CUBLAS.jl copied to clipboard

Allow switching devices. Use `__init__`.

Open lucasb-eyer opened this issue 9 years ago • 3 comments

This introduces the use_current_device function which initializes CUBLAS for the currently active device, thus allowing the use of blas.jl functions on non-default GPU.

In addition, move initialization (and registration of the finalizer) into __init__ so that we can make use of pre-compilation in the future.

The default behavior should be unchanged by this.

lucasb-eyer avatar Oct 24 '15 12:10 lucasb-eyer

Hi @lucasb-eyer, I'm sorry it took so long for someone to look at this. This PR LGTM and I'll merge it tonight if no one has any objections.

kshyatt avatar Nov 23 '15 23:11 kshyatt

No worries, I'm not in a hurry.

lucasb-eyer avatar Nov 25 '15 08:11 lucasb-eyer

ping, I guess =)

lucasb-eyer avatar Dec 01 '15 12:12 lucasb-eyer