Skip to content

Commit

Permalink
Update examples/UDF-Examples/RAPIDS-accelerated-UDFs/README.md
Browse files Browse the repository at this point in the history
Co-authored-by: Jason Lowe <[email protected]>
  • Loading branch information
nvliyuan and jlowe authored Sep 26, 2023
1 parent aa3ab06 commit 783afe8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/UDF-Examples/RAPIDS-accelerated-UDFs/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ Below are some showcases about implementing RAPIDS accelerated scala UDF by JNI
If there is no existing simple Java API we could leverage, we can write native custom code.
The Java class for the UDF is similar as the previous URLDecode/URLEncode demo, we need to implement a cosineSimilarity
function in C++ code and goes into the native code as quickly as possible, because it is easier to write the code
safely. In the native code, it `reinterpret_cast` the input to columnview, do some sanity checking and convert to list
safely. In the native code, it `reinterpret_cast` the input to a column view, do some sanity checking and convert to list
column views, then compute the cosine similarity, finally return the unique pointer to a column, release the underlying resources.
On Java side we are going to wrap it in a columnvector and own that resource. In `cosine_similarity.cu` we implement
the computation as the actual CUDA kernel. In CUDA kernel part, we can leverage thrust template library to write the
Expand Down

0 comments on commit 783afe8

Please sign in to comment.