Support different codebook sizes in CUDA kernels
dryman opened this issue · comments
Does this package support other dimensions than 128?
We'd like to test it on GloVe, which is 100.
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 veccost
that you linked above is not called from the Julia part. veccost2
is
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.