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

[CUSOLVER] Interface gesv and gels #2406

Draft
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

amontoison
Copy link
Member

  • Interface gesv and gels for solving dense linear problems AX =B and min ||B - AX||.
    These routines are quite complex because they can use low precision to perform some operations and provide many options to the user.
  • I also created a file helpers.jl to store all structures for CUSOLVER in one file. We already did that in lib/cusparse.

@amontoison amontoison marked this pull request as draft June 2, 2024 05:37
return CUSOLVER_R_16F
elseif T === BFloat16
return CUSOLVER_R_16BF
# elseif T === TensorFloat32
Copy link
Member Author

Choose a reason for hiding this comment

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

@maleadt Do we have a Julia type for "TensorFloat32"?

Copy link
Member

Choose a reason for hiding this comment

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

No. AFAIU, TensorFloat32 isn't an actual type that can be used in kernels, but only through tensor cores.

@maleadt maleadt added cuda libraries Stuff about CUDA library wrappers. enhancement New feature or request labels Sep 18, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cuda libraries Stuff about CUDA library wrappers. enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants