Skip to content
This repository has been archived by the owner on Nov 25, 2024. It is now read-only.

Commit

Permalink
add parameter explanation for user_defined_sms
Browse files Browse the repository at this point in the history
  • Loading branch information
linhu-nv committed Feb 1, 2024
1 parent e4f361a commit d3316e8
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions cpp/include/wholememory/embedding.h
Original file line number Diff line number Diff line change
Expand Up @@ -130,6 +130,7 @@ wholememory_error_code_t wholememory_destroy_embedding_cache_policy(
* @param memory_location : Memory Location of the underlying WholeMemory
* @param optimizer : Optimizer to use for training, if don't train embedding, use nullptr
* @param cache_policy : Cache policy for this embedding, if don't use cache, use nullptr
* @param user_defined_sms : User-defined sms number for raw embedding gather/scatter
* @return : wholememory_error_code_t
*/
wholememory_error_code_t wholememory_create_embedding(
Expand Down

0 comments on commit d3316e8

Please sign in to comment.