Skip to content

Commit 049c30e

Browse files
authored
[oneMKL] [BLAS] Add half/bfloat16 support for select L1 APIs (#422)
1 parent 90bd028 commit 049c30e

5 files changed

Lines changed: 18 additions & 0 deletions

File tree

source/elements/oneMKL/source/domains/blas/axpy.rst

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,8 @@ where:
3232
:header-rows: 1
3333

3434
* - T
35+
* - ``half``
36+
* - ``bfloat16``
3537
* - ``float``
3638
* - ``double``
3739
* - ``std::complex<float>``

source/elements/oneMKL/source/domains/blas/dot.rst

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,10 @@ The ``dot`` routines perform a dot product between two vectors:
2626

2727
* - T
2828
- T_res
29+
* - ``half``
30+
- ``half``
31+
* - ``bfloat16``
32+
- ``bfloat16``
2933
* - ``float``
3034
- ``float``
3135
* - ``double``

source/elements/oneMKL/source/domains/blas/nrm2.rst

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,10 @@ where:
3030

3131
* - T
3232
- T_res
33+
* - ``half``
34+
- ``half``
35+
* - ``bfloat16``
36+
- ``bfloat16``
3337
* - ``float``
3438
- ``float``
3539
* - ``double``

source/elements/oneMKL/source/domains/blas/rot.rst

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,10 @@ the sum of two of these scalar-vector products as follow:
3535

3636
* - T
3737
- T_scalar
38+
* - ``half``
39+
- ``half``
40+
* - ``bfloat16``
41+
- ``bfloat16``
3842
* - ``float``
3943
- ``float``
4044
* - ``double``

source/elements/oneMKL/source/domains/blas/scal.rst

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,10 @@ where:
3232

3333
* - T
3434
- T_scalar
35+
* - ``half``
36+
- ``half``
37+
* - ``bfloat16``
38+
- ``bfloat16``
3539
* - ``float``
3640
- ``float``
3741
* - ``double``

0 commit comments

Comments
 (0)