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

Add Support for BFloat16 #298

Open
tgymnich opened this issue Mar 4, 2024 · 3 comments
Open

Add Support for BFloat16 #298

tgymnich opened this issue Mar 4, 2024 · 3 comments
Labels
enhancement New feature or request good first issue Good for newcomers help wanted Extra attention is needed

Comments

@tgymnich
Copy link
Member

tgymnich commented Mar 4, 2024

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 New feature or request 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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request good first issue Good for newcomers help wanted Extra attention is needed
Projects
None yet
Development

No branches or pull requests

3 participants