squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats 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_HOST_DEVICE
cutlass::bfloat16_t abs(cutlass::bfloat16_t const& h) { 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 CUTLASS_HOST_DEVICE