NVIDIA/cutlass
1
0
Fork 0
mirror of https://github.com/NVIDIA/cutlass.git synced 2026-05-12 01:10:08 +00:00
Code Issues 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