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

fix gen rand data and use of cublasHgemm #1

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

fix gen rand data and use of cublasHgemm #1

wants to merge 1 commit into from

Conversation

zxy-zzz
Copy link

@zxy-zzz zxy-zzz commented Aug 14, 2024

你好，我发现你用的数据生成方法，生成的数据太大，gemm乘法的结果都是inf，你可以打印看看，最终导致两个inf作比较，结果是0，这是有问题的。
这个PR纠正了数据生成方法，更换了cuBLAS中HGEMM的调用方式。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@zxy-zzz