-
Notifications
You must be signed in to change notification settings - Fork 945
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
MAHOUT-1974 CUDA support #310
base: CUDA
Are you sure you want to change the base?
Conversation
andy@micheal:~/sandbox/mahout_cuda/cuda$ export SIZE_N=5000 andy@micheal:~/sandbox/mahout_cuda/cuda$ export SIZE_M=5000 andy@micheal:~/sandbox/mahout_cuda/cuda$ export SIZE_S=5000 andy@micheal:~/sandbox/mahout_cuda/cuda$ export DENSITY=.2 andy@micheal:~/sandbox/mahout_cuda/cuda$ export NUM_RUNS=2 andy@micheal:~/sandbox/mahout_cuda/cuda$ export SEED=1234
… into AP-UNIT-TEST
Tests pass on my system:
|
@nsakharnykh @rawkintrevo I intend to have |
@nsakharnykh , @rawkintrevo, I ran out of time tonight to finish out I pushed my beginning work up to my MAHOUT-1974 branch. Nothing really worth looking at right now, but I wil' make a PR against this when I get the Regardless, I should have at least a quick n dirty version ready to go soon, while i work out what we'll need for experiments and benchmarking. We can still discuss and consider different SPARK configurations tomorrow with out As I mentioned on the last call we allow a "Sparse" DRM's in-core components to be both sparse and dense. Currently the threshold for conversion of a DRM block to be changed from a sparse to a dense matrix is pretty high (25% non zero estimate). In the future we will need to allow the user to set the sparsity somehow. |
@andrewpalumbo regarding column-major: yes, this is the default mode for CUBLAS, sorry I think I didn't mention it in my original email. There are a couple options we can exercise here. 1. We can use transposed versions of |
cuda/pom.xml
Outdated
<parent> | ||
<groupId>org.apache.mahout</groupId> | ||
<artifactId>mahout</artifactId> | ||
<version>0.13.0-SNAPSHOT</version> |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
needs to be 0.13.1-SNAPSHOT
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done
@nsakharnykh I have my MAHOUT-1974 branch that is almost complete with dense, etc (less the column major issues. We'd discussed just making a PR against this. but It may be easiest if you just went ahead and pushed this to MAHOUT/CUDA, and then I'll make a PR against that, which will be public so that others may comment on it. |
@nsakharnykh https://github.com/andrewpalumbo/mahout/tree/MAHOUT-1974/cuda ^^ |
@andrewpalumbo Ok, sounds good. I'll try to push what I have as soon as I have some time in front of my laptop. I'm currently at GTC so my schedule is a bit fragmented. |
Great, thanks. I figured you were there, and very busy, I'll keep working on my end, and there should be no (or few conflicts).. no rush, since my branch is based off of yours. |
looking awesome @nsakharnykh @andrewpalumbo Before merging, don't forget to fill out |
@rawkintrevo I asked @nsakharnykh to just go ahead and push this to the mahout/CUDA branch, since he's already up at GTC, and we're pushing this through as quickly as possible, and has spotty time to do this. I will immediately open up a [WIP] PR from my https://github.com/andrewpalumbo/mahout/tree/MAHOUT-1974/cuda branch (on top of his) and will fill out the md from there. |
Just checking if we need to keep this PR open - I'm guessing this is already merged in feature branch: https://github.com/apache/mahout/tree/CUDA |
Initial PR for CUDA bindings support through JCuda