fix argument name in comment

rapidsai · Aug 2, 2024 · e661c4f · e661c4f
1 parent b14440a
commit e661c4f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cpp/include/wholememory/wholememory_tensor.h b/cpp/include/wholememory/wholememory_tensor.h
@@ -143,7 +143,7 @@ wholememory_error_code_t wholememory_tensor_get_entry_offsets(
 
 /**
  * Get entry count of each rank from WholeMemory Tensor
- * @param entry_offsets : returned entry count of each rank
+ * @param entry_partition : returned entry count of each rank
  * @param wholememory_tensor : WholeMemory Tensor
  * @return : wholememory_error_code_t
  */

diff --git a/cpp/src/wholememory_ops/functions/embedding_cache_func.h b/cpp/src/wholememory_ops/functions/embedding_cache_func.h
@@ -55,7 +55,7 @@ wholememory_error_code_t update_cache_direct_same_comm(
  * @param wm_raw_memory_embedding : the WholeMemory Tensor that is to be cached which stores all
  * embeddings.
  * @param cache_comm : communicator of cache
- * @param embedding_entry_count_per_cache_rank : embedding entries covered by each cache rank
+ * @param embedding_entry_offsets : embedding entry offset of each cache rank
  * @param cache_local_data : embedding_cache_local_data of wm_raw_memory_embedding
  * @param cache_set_coverage : cache set coverage
  * @param p_env_fns : env fns