Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Cheqi/merge gpu code #159

Merged
merged 69 commits into from Aug 10, 2020
Merged

Cheqi/merge gpu code #159

merged 69 commits into from Aug 10, 2020

Conversation

MaggieQi
Copy link
Member

@MaggieQi MaggieQi commented Aug 7, 2020

No description provided.

Qi Chen and others added 30 commits December 19, 2019 16:20
* try to add an end-to-end tutorial for building vector search online s… (#116)

* try to add an end-to-end tutorial for building vector search online service.

* update the tutorial and getting start page

* fix space and new line issues

* fix gcc5 compiling issue (#120)

* fix gcc5 compiling issue

* fix gcc5 compiling issue
bkarsin and others added 21 commits June 30, 2020 23:19
* Small changes before new code...

* Fix cmake to work on dtcluster

* Debugging and testing

* Fixed several bugs, added unit test for internal testing, just need to fix parallel RNG pruning.

* Initial GPU refinement code

* Fixed bug with refinement

* Fixed bug with RNG shrinking function

* Stable GPU refinement debugged.

* Cleaned up code a bit

* Removed specific paths for local build

* Initial implementation of "strict" initial RNG construction

* Fixed strict RNG construction and added separate options for loose and strict construction

* Initial code of new strict RNG construction that reaches refinement accuracy target.

* Fix / improve int8 cosine using dp4a intrinsic

* Debugging batched execution of RNG graph build

* before merge

* More work debugging TPT issues for large datasets

* Polishing, fixing default values, and removing some scaffolding

* before merge

* Added more useful error handling and debug info

Co-authored-by: Ben Karsin <bkarsin@dtlogin.cm.cluster>
Co-authored-by: Ben Karsin <bkarsin@dt03.eth.cluster>
Co-authored-by: Ben Karsin <bkarsin@prm-dgx-17.nvidia.com>
Co-authored-by: Ben Karsin <bkarsin@prm-dgx-01.nvidia.com>
Co-authored-by: Ben Karsin <bkarsin@prom.nvidia.com>
Co-authored-by: Ben Karsin <bkarsin@prm-dgx-23.nvidia.com>
Co-authored-by: Ben Karsin <bkarsin@prm-dgx-05.nvidia.com>
Co-authored-by: Ben Karsin <bkarsin@prm-dgx-04.nvidia.com>
Co-authored-by: Ben Karsin <bkarsin@prm-dgx-12.nvidia.com>
Co-authored-by: Ben Karsin <bkarsin@prm-dgx-15.nvidia.com>
…run on hardware with compute capability < 61 (#153)

Co-authored-by: Ben Karsin <bkarsin@dtlogin.cm.cluster>
* Added non-intrinsic version of cosine distance so it can be compiled/run on hardware with compute capability < 61

* Add signed int8 support

Co-authored-by: Ben Karsin <bkarsin@dtlogin.cm.cluster>
Co-authored-by: Ben Karsin <bkarsin@prom.nvidia.com>
{
std::lock_guard<std::mutex> lock(m_dataUpdateLock[node]);

SizeType* nodes = m_pNeighborhoodGraph[node];

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

check node overflow?

@MaggieQi MaggieQi merged commit bc02f16 into master Aug 10, 2020
@MaggieQi MaggieQi deleted the cheqi/merge_gpu_code branch August 11, 2020 08:29
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants