GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-09-23 14:53:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Disable array BF16 to F32 conversions in Power

Browse Source
This commit is contained in:
Chip Kerchner 2023-02-10 20:06:58 +00:00 committed by Rasmus Munk Larsen
parent c999284bad
commit 0ecae61568
1 changed files with 1 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
Eigen/src/Core/arch/AltiVec/MatrixProductMMAbfloat16.h
View File

@ -189,27 +189,11 @@ void gemmMMAbfloat16(const DataMapper& res, const bfloat16* blockA, const bfloat
ei_declare_aligned_stack_constructed_variable(float, result, cols*rows, 0); ei_declare_aligned_stack_constructed_variable(float, result, cols*rows, 0);
typedef typename DataMapper::LinearMapper LinearMapper; typedef typename DataMapper::LinearMapper LinearMapper;
Packet4f z = pset1<Packet4f>(float(0));
for(Index j = 0; j < cols; j++){ for(Index j = 0; j < cols; j++){
const LinearMapper res2 = res.getLinearMapper(0, j); const LinearMapper res2 = res.getLinearMapper(0, j);
float *result2 = result + j*rows; float *result2 = result + j*rows;
Index i = 0;
for(; i + 32 <= rows; i+=32){
Packet4f r32_0 = reinterpret_cast<Packet4f>(res2.template loadPacket<Packet8bf>(i + 0).m_val);
Packet4f r32_1 = reinterpret_cast<Packet4f>(res2.template loadPacket<Packet8bf>(i + 8).m_val);
Packet4f r32_2 = reinterpret_cast<Packet4f>(res2.template loadPacket<Packet8bf>(i + 16).m_val);
Packet4f r32_3 = reinterpret_cast<Packet4f>(res2.template loadPacket<Packet8bf>(i + 24).m_val);
pstore(result2 + i + 0, vec_mergeo(r32_0, z));
pstore(result2 + i + 4, vec_mergee(r32_0, z));
pstore(result2 + i + 8, vec_mergeo(r32_1, z));
pstore(result2 + i + 12, vec_mergee(r32_1, z));
pstore(result2 + i + 16, vec_mergeo(r32_2, z));
pstore(result2 + i + 20, vec_mergee(r32_2, z));
pstore(result2 + i + 24, vec_mergeo(r32_3, z));
pstore(result2 + i + 28, vec_mergee(r32_3, z));
}
BFLOAT16_UNROLL BFLOAT16_UNROLL
for(; i < rows; i++){ for(Index i = 0; i < rows; i++){
result2[i] = res2(i); result2[i] = res2(i);
} }
} }