devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Preliminary HIP bfloat16 GPU support.

Browse Source
This commit is contained in:
Rohit Santhanam
2021-10-27 18:36:45 +00:00
committed by Antonio Sánchez
parent 40bbe8a4d0
commit 48e40b22bf
4 changed files with 621 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Eigen/src/Core/util/ConfigureVectorization.h
View File

@@ -468,6 +468,8 @@
#include <hip/hip_vector_types.h>
#define EIGEN_HAS_HIP_FP16
#include <hip/hip_fp16.h>
#define EIGEN_HAS_HIP_BF16
#include <hip/hip_bfloat16.h>
#endif