Tamas Bela Feher

[FEA] Support index building from host dataset for IVF-FLAT

1

**Is your feature request related to a problem? Please describe.** Currently only device datasets are allowed for extending the index (which is part of the build process). The goal is...

Add raft-ann-bench script to generate synthetic dataset

2

This PR adds a Python script to generate synthetic dataset.

C++ API test with synthetic data

1

Test program to run TF-TRT conversion with the C++API with synthetic data. This is based on #271 Todo - [ ] rename / clean up files - [ ] script...

A converted model contains the frozen variables (const), the original variables, and the variables saved into the TRT engine as weights. This can lead up to 3x size of the...

BERT Large (TF2) model conversion fails

5

The following code loads BERT Large model from TF Hub, and tries to convert using TF-TRT. ```python import os os.environ["TF_CPP_VMODULE"]="trt_engine_utils=2,trt_engine_op=2,convert_nodes=2,convert_graph=2,segment=2,trt_shape_optimization_profiles=2,trt_engine_resource_ops=2" import tensorflow as tf import tensorflow_hub as hub # ##...

Shape output bug in dynamic shape mode

2

If we enable dynamic shape mode, and any of the TRT segments has an output from a shape op, then conversion crashes the application. This problem occurs in practice for...

Random sampling of training set for IVF methods was reverted in https://github.com/rapidsai/raft/pull/2144 due to the large memory usage of the subsample method. Since then, PR https://github.com/rapidsai/raft/pull/2155 has implemented a new...

The pointer based APIs in RAFT for `ivf_pq::build` and `ivf_pq::extend` do accept host and device arrays (as well es pinned, and managed arrays). Accepting host arrays is an important feature:...

[BUG] Multi GPU KMeans memory usage is 2x larger than expected.

19

**Describe the bug** The expected GPU memory usage of cuML's kmeans algorithm is `(n_rows * n_cols + n_cluster*n_cols) * sizeof(MathT)`, where `(n_rows, n_cols)` is the shape of the input matrix....

**Describe the bug** The [raft logging macros rely](https://github.com/rapidsai/raft/blob/0e68cfac7ff82009724c8e4e747b7ca9cd5f664a/cpp/include/raft/core/logger-macros.hpp#L88C21-L88C30) on `RAFT_NAME` static variable to get the logger. `RAFT_NAME` is defined in the `raft` namespace, but the macros use the variable without...