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

Add Support for BFloat16 #298

Open
tgymnich opened this issue Mar 4, 2024 · 3 comments · May be fixed by #446
Open

Add Support for BFloat16 #298

tgymnich opened this issue Mar 4, 2024 · 3 comments · May be fixed by #446
Labels
good first issue Good for newcomers help wanted Extra attention is needed

Comments

@tgymnich
Copy link
Member

tgymnich commented Mar 4, 2024 •
edited
Loading

With Julia 1.11 coming up, we will have native support for BFloat16 JuliaLang/julia#51470.

Metal also supports BFloat16 onwards from Apple6 GPU architecture.

M1 belongs to the Apple7 GPU family

https://developer.apple.com/documentation/metalperformanceshaders/mpsdatatype
@tgymnich tgymnich added enhancement help wanted Extra attention is needed good first issue Good for newcomers labels Mar 4, 2024
@Dsantra92
Copy link

Hi, I would love to work on this issue.

@Dsantra92
Copy link

Looks like precompilation fails for Metal.jl in 1.11 alpha 2 because of a version restriction in GPUCompiler.jl code.

Also there are multiple failures with current alpha if we ignore the code component for GPUCompiler.CodeCache like Metal.rand. Also the Core.BFloat16 is only partially implemented in 1.11 for tricking LLVM I suppose. For full usage we have to rely on BFloat16s.jl. I feel I should wait for 1.11 stable before starting working on this issue.

@ViralBShah
Copy link
Contributor

I suspect the version restriction is in place because the GPUCompiler probably needs to be updated for 1.11.

I believe the idea was to have minimal support in Julia base, and use packages to implement the full functionality, which can move much faster with releases and such.

@christiangnrd christiangnrd linked a pull request Oct 4, 2024 that will close this issue
Draft
2 tasks
@christiangnrd christiangnrd mentioned this issue Oct 12, 2024
Open
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
good first issue Good for newcomers help wanted Extra attention is needed
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

Bfloat16 support JuliaGPU/Metal.jl
3 participants
@ViralBShah @tgymnich @Dsantra92