Skip to content

Conversation

@mtamburrano
Copy link
Contributor

caffe_set for GPU mode.

I'm not sure this works when alpha != 0, I've not at the moment a gpu to test it. My doubt is that when the vector Y is gpu data, could not be directly accessible.
If I have no directions in the next two days, I'll try to test it with a gpu.

@mtamburrano mtamburrano mentioned this pull request Dec 1, 2014
@sguada
Copy link
Contributor

sguada commented Dec 2, 2014

Not needed there is a caffe_gpu_set

@longjon
Copy link
Contributor

longjon commented Dec 2, 2014

Right, use caffe_gpu_set; it uses a proper kernel. (If I recall correctly, the reason there isn't a corresponding caffe_gpu_copy is because caffe_copy works with UVA for any kind of copy.)

@longjon longjon closed this Dec 2, 2014
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants