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.

Sign up for GitHub

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

Jump to bottom

[FEA] Calculating distances with FP32 in NN Descent #195

Open
jinsolp opened this issue Jun 18, 2024 · 0 comments
Open

[FEA] Calculating distances with FP32 in NN Descent #195

jinsolp opened this issue Jun 18, 2024 · 0 comments
Labels
feature request New feature or request

Comments

@jinsolp
Copy link

jinsolp commented Jun 18, 2024

I hope NN Descent can support calculating distances in local_join_kernel using fp32 instead of fp16 (__half).

Due to precision issues, it is difficult to grab the distances calculated by NN Descent and compare it against fp32-calculated distances for additional operations outside of NN Descent such as making predictions.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
feature request New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@jinsolp