extern

Author: KyleFromNVIDIA

cpp/src/neighbors/detail/cagra/jit_lto_kernels/compute_distance_kernel.cu.in

@@ -14,6 +14,6 @@ template __device__ @distance_type@ compute_distance<@team_size@, @dataset_block
 }  // namespace cuvs::neighbors::cagra::detail
 
 extern "C" {
-__device__ @distance_type@ (*const compute_distance_ptr)(const typename cuvs::neighbors::cagra::detail::dataset_descriptor_base_t<@data_type@, @index_type@, @distance_type@>::args_t, @index_type@) =
+extern __device__ @distance_type@ (*const compute_distance_ptr)(const typename cuvs::neighbors::cagra::detail::dataset_descriptor_base_t<@data_type@, @index_type@, @distance_type@>::args_t, @index_type@) =
     &cuvs::neighbors::cagra::detail::compute_distance<@team_size@, @dataset_block_dim@, @pq_bits@, @pq_len@, @codebook_type@, @data_type@, @index_type@, @distance_type@, @query_type@>;
 }

cpp/src/neighbors/detail/cagra/jit_lto_kernels/setup_workspace_kernel.cu.in

@@ -13,7 +13,7 @@ template __device__ cuvs::neighbors::cagra::detail::dataset_descriptor_base_t<@d
 }  // namespace cuvs::neighbors::cagra::detail
 
 extern "C" {
-__device__ cuvs::neighbors::cagra::detail::dataset_descriptor_base_t<@data_type@, @index_type@, @distance_type@>* (*const setup_workspace_ptr)(
+extern __device__ cuvs::neighbors::cagra::detail::dataset_descriptor_base_t<@data_type@, @index_type@, @distance_type@>* (*const setup_workspace_ptr)(
     cuvs::neighbors::cagra::detail::dataset_descriptor_base_t<@data_type@, @index_type@, @distance_type@>*, void*, const @data_type@*, uint32_t) =
     &cuvs::neighbors::cagra::detail::setup_workspace<@team_size@, @dataset_block_dim@, @pq_bits@, @pq_len@, @codebook_type@, @data_type@, @index_type@, @distance_type@, @query_type@>;
 }