Skip to content

Commit

Permalink
Remove preferred_thread_block_size search param, since it will be dep…
Browse files Browse the repository at this point in the history
…recated in rapidsai#926
  • Loading branch information
tfeher committed Nov 11, 2022
1 parent bc4c66c commit 52feeac
Showing 1 changed file with 0 additions and 7 deletions.
7 changes: 0 additions & 7 deletions python/pylibraft/pylibraft/neighbors/ivf_pq.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -76,7 +76,6 @@ cdef extern from "raft/neighbors/ivf_pq_types.hpp" \
uint32_t n_probes
cudaDataType_t lut_dtype
cudaDataType_t internal_distance_dtype
uint32_t preferred_thread_block_size

cdef extern from "raft/neighbors/specializations/ivf_pq_specialization.hpp" \
namespace "raft::neighbors::ivf_pq":
Expand Down Expand Up @@ -448,10 +447,6 @@ class IvfPq:
internal_distance_dtype: default = CUDA_R_32F (float)
Storage data type for distance/similarity computation.
Possible values [CUDA_R_32F, CUDA_R_16F]
preferred_thread_block_size : int, default = 0
Thread block size of the distance calculation kernel at search time.
When zero, an optimal block size is selected using a heuristic.
Possible values: [0, 256, 512, 1024]
Returns
-------
Expand All @@ -476,7 +471,6 @@ class IvfPq:
assert(n_queries > 0)
assert(queries_dt in [np.dtype('float32'), np.dtype('byte'), np.dtype('ubyte') ])


assert(dim_queries == self._dim)

neighbors_cai = neighbors.__cuda_array_interface__
Expand All @@ -495,7 +489,6 @@ class IvfPq:
params.n_probes = n_probes
# params.lut_dtype = lut_dtype
# params.internal_distance_dtype = internal_distance_dtype
params.preferred_thread_block_size = preferred_thread_block_size

cdef handle_t* handle_ = <handle_t*><size_t>self.handle.getHandle()
cdef index[uint64_t] *idx = <index[uint64_t]*><uintptr_t>self._index
Expand Down

0 comments on commit 52feeac

Please sign in to comment.