arrayfire / arrayfire/arrayfire

Add support for bf16 for afoneapi/afcuda

Open
#3,470 0 comments 0 reactions 0 assignees View on GitHub
feature
Dominant language
C++
Stars
4.9k
Forks
555
Avg merge
1h 24m
Merged PRs (30d)
1

Description

Add suppor for bf16 datatype
https://github.com/arrayfire/arrayfire/blob/master/include/af/defines.h
for :
- afcuda/nvidia gpus
https://sources.debian.org/src/nvidia-cuda-toolkit/11.2.2-3%2Bdeb11u3/arm64/cuda_cudart/targets/sbsa-linux/include/cuda_bf16.hpp/#L126

- 4th gen intel CPUs (SaphirreRapids) with avx512bf16 / AMX

Description
===========
* Additional information about the feature you would like to add
bf16 datatype
* What problem are you trying to solve?
computation at bf16 precision
* (Optional) API of new function
No new API, just a new type in defines.h

* (Optional)Are there other libraries that implement this feature?
pytorch, numpy, oneDNN, ...

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.