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

NVIDIA · Nov 13, 2023 · 1ab6cc7 · 1ab6cc7
1 parent 5ae8133
commit 1ab6cc7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/include/cutlass/bfloat16.h b/include/cutlass/bfloat16.h
@@ -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