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

2023-11-13 13:29:45 -05:00
parent 5ae8133cfa
commit 1ab6cc7b68
1 changed files with 1 additions and 1 deletions
@@ -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