注意

RAFT 中的向量搜索和聚类算法正在迁移到专门用于向量搜索的新库 cuVS。在此迁移期间,我们将继续支持 RAFT 中的向量搜索算法,但在 RAPIDS 24.06 (六月) 版本之后将不再更新它们。我们计划在 RAPIDS 24.10 (十月) 版本之前完成迁移,并在 24.12 (十二月) 版本中将它们从 RAFT 中完全移除。

NVTX#

#include <raft/core/nvtx.hpp>

namespace raft::core

template<typename Domain = domain::app, typename ...Args>
inline void raft::common::nvtx::push_range(const char *format, Args... args)#

推送一个命名 NVTX 范围。

模板参数:

Domain – 可选结构体,定义 NVTX 域消息;您可以按如下方式创建带有自定义消息的新域

struct custom_domain { static constexpr char const* name{"custom message"}; }
注意: 确保对 push_rangepop_range 使用相同的域。

参数:
  • format – 范围名称格式 (接受 printf 风格的参数)

  • args – 用于 printf 风格格式化的参数

template<typename Domain = domain::app>
inline void raft::common::nvtx::pop_range()#

弹出最新的范围。

模板参数:

Domain – 可选结构体,定义 NVTX 域消息;您可以按如下方式创建带有自定义消息的新域

struct custom_domain { static constexpr char const* name{"custom message"}; }
注意: 确保对 push_rangepop_range 使用相同的域。

template<typename Domain = domain::app>
class range#
#include <nvtx.hpp>

推送一个命名 NVTX 范围,该范围将在对象生命周期结束时弹出。

请参阅 Usage 获取使用示例。

模板参数:

Domain – 可选结构体,定义 NVTX 域消息;您可以按如下方式创建带有自定义消息的新域

struct custom_domain { static constexpr char const* name{"custom message"}; }

公共函数

template<typename ...Args>
inline explicit range(const char *format, Args... args)#

推送一个命名 NVTX 范围。在对象生命周期结束时,弹回该范围。

参数:
  • format – 范围名称格式 (接受 printf 风格的参数)

  • args – 用于 printf 风格格式化的参数