-
Notifications
You must be signed in to change notification settings - Fork 4
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
Support different codebook sizes in CUDA kernels #39
Comments
Yep, just add your data reading functions to Rayuela.jl/src/read_datasets.jl Line 4 in ccf22ab
|
Does the following code in cuda matter? Rayuela.jl/deps/src/cudautils.cu Lines 93 to 95 in ccf22ab
|
Oh right, sorry. The GPU part is hardcoded. |
I checked again, since I definitely ran this code on the GPU for my thesis. The function Rayuela.jl/deps/src/cudautils.cu Lines 154 to 155 in ccf22ab
which only hardcodes the number of entries in the codebook (not the dataset dimensionality). I'm changing the name of this issue to reflect that. |
Also opened #41 to keep track of code deletion. |
Does this package support other dimensions than 128?
We'd like to test it on GloVe, which is 100.
The text was updated successfully, but these errors were encountered: