Skip to content

[BUG]RAFT Failure error while running the cuVS Python SQ API #623

@AmeliaYe

Description

@AmeliaYe

Describe the bug
The original embedding before SQ works on cagra, but after SQ, I am encountering this error. Does cagra.build support int8? Also after quantization, the embedding is int8, containing negative number, not sure if it suppose to be int8 or uint8?

Image

Image

Steps/Code to reproduce bug
Link to the notebook with SQ and failed cagra.build: https://gitlab-master.nvidia.com/ameliay/cagra-umap/-/commit/7b0739622aff28c192ed49887c0327ea733390ac#c187ee3d6e010901b2354522adb927344250e1bd_0_291

error message also there.

Expected behavior
cagra.build runs without error so that we can continue to cagra.search and calculate recall score

Environment details (please complete the following information):

  • Environment location: Bare-metal
  • Method of RAFT install: conda
    • If method of install is [Docker], provide docker pull & docker run commands used

Additional context
Add any other context about the problem here.

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions