JuliaAttic / CUBLAS.jl

Julia interface to CUBLAS
Other
26 stars 19 forks source link

Allow switching devices. Use `__init__`. #15

Open lucasb-eyer opened 9 years ago

lucasb-eyer commented 9 years ago

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.

kshyatt commented 8 years ago

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.

lucasb-eyer commented 8 years ago

No worries, I'm not in a hurry.

lucasb-eyer commented 8 years ago

ping, I guess =)