We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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
Support fp16 operations for raft::statt::mean()
raft::statt::mean()
Describe the solution you'd like
fp16 support where the intermediate results are accumulated in fp32, to avoid potential overflow.
Additional context
This issue came up in rapidsai/cuvs#716, and resulted in the following workaround
https://github.com/rapidsai/cuvs/pull/716/files#diff-15fe794d70734cd604f7fda31ee8c2240839939c1c811df6cceb71752a0c35c8R242
The text was updated successfully, but these errors were encountered:
No branches or pull requests
Support fp16 operations for
raft::statt::mean()
Describe the solution you'd like
fp16 support where the intermediate results are accumulated in fp32, to avoid potential overflow.
Additional context
This issue came up in rapidsai/cuvs#716, and resulted in the following workaround
https://github.com/rapidsai/cuvs/pull/716/files#diff-15fe794d70734cd604f7fda31ee8c2240839939c1c811df6cceb71752a0c35c8R242
The text was updated successfully, but these errors were encountered: