注意
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_range
和pop_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_range
和pop_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"}; }