Skip to content

Commit 9069280

Browse files
committed
fix tests
1 parent 77114a4 commit 9069280

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

cpp/src/detail/jit_lto/AlgorithmPlanner.cu

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@ std::shared_ptr<AlgorithmLauncher> AlgorithmPlanner::get_launcher()
5353
static std::mutex cache_mutex;
5454
std::lock_guard<std::mutex> lock(cache_mutex);
5555
if (launchers.count(launch_key) == 0) {
56+
RAFT_LOG_INFO("A first time JIT compilation has been triggered for your algorithm.");
5657
add_entrypoint();
5758
add_device_functions();
5859
std::string log_message =
@@ -61,7 +62,7 @@ std::shared_ptr<AlgorithmLauncher> AlgorithmPlanner::get_launcher()
6162
log_message += device_function + ",";
6263
}
6364
log_message.pop_back();
64-
RAFT_LOG_INFO("%s", log_message.c_str());
65+
RAFT_LOG_DEBUG("%s", log_message.c_str());
6566
launchers[launch_key] = this->build();
6667
}
6768
return launchers[launch_key];

cpp/src/neighbors/ivf_flat/ivf_flat_interleaved_scan_jit.cuh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,7 @@ void launch_kernel(const index<T, IdxT>& index,
190190
std::string metric_name = "metric_udf_" + udf_hash;
191191
auto& nvrtc_lto_compiler = nvrtc_compiler();
192192
std::string key =
193-
metric_name + "_" + std::to_string(Veclen) + "_" +
193+
metric_name + "_veclen_" + std::to_string(Veclen) + "_" +
194194
make_fragment_key<decltype(get_data_type_tag<T>()), decltype(get_acc_type_tag<AccT>())>();
195195
nvrtc_lto_compiler.compile(key, metric_udf);
196196
kernel_planner.template add_metric_device_function<decltype(get_data_type_tag<T>()),

0 commit comments

Comments
 (0)