youngkingdom/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix std::abs overloading for bfloat16_t (#1179)

Browse Source
This commit is contained in:
Changho Hwang
2023-11-14 02:29:45 +08:00
committed by GitHub
parent 5ae8133cfa
commit 1ab6cc7b68
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/cutlass/bfloat16.h
View File

@ -212,7 +212,7 @@ bool signbit(cutlass::bfloat16_t const& h) {
CUTLASS_HOST_DEVICE
cutlass::bfloat16_t abs(cutlass::bfloat16_t const& h) {
return cutlass::bfloat16_t::bitcast(h.raw() & 0x7fffffff);
return cutlass::bfloat16_t::bitcast(h.raw() & 0x7fff);
}
CUTLASS_HOST_DEVICE