Skip to content

Commit

Permalink
Fix docs builds
Browse files Browse the repository at this point in the history
  • Loading branch information
bdice committed Jan 31, 2025
1 parent 19b8103 commit 23ba694
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 3 deletions.
6 changes: 4 additions & 2 deletions cpp/include/raft/random/device/sample.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -27,12 +27,14 @@ namespace raft::random::device {

/**
* @brief warp-level random sampling of an index.
*
* It selects an index with the given discrete probability
* distribution(represented by weights of each index)
* distribution(represented by weights of each index).
* Only thread 0 will contain the valid reduced result.
*
* @param rng random number generator, must have next_u32() function
* @param weight weight of the rank/index.
* @param idx index to be used as rank
* @return only the thread0 will contain valid reduced result
*/
template <typename T, typename rng_t, typename i_t = int>
DI void warp_random_sample(rng_t& rng, T& weight, i_t& idx)
Expand Down
2 changes: 1 addition & 1 deletion cpp/include/raft/util/itertools.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ namespace raft::util::itertools {
* fields of the structure (if the structure has more fields, some might be initialized
* with their default value).
* @param lists One or more initializer lists.
* @return std::vector<S> A vector of structures containing the cartesian product.
* @return `std::vector<S>` A vector of structures containing the cartesian product.
*/
template <typename S, typename... Args>
std::vector<S> product(std::initializer_list<Args>... lists)
Expand Down

0 comments on commit 23ba694

Please sign in to comment.