diff --git a/third_party/xla/xla/ffi/attribute_map.cc b/third_party/xla/xla/ffi/attribute_map.cc index 70cce65d9d7cb7..7c14ecfd49d742 100644 --- a/third_party/xla/xla/ffi/attribute_map.cc +++ b/third_party/xla/xla/ffi/attribute_map.cc @@ -47,20 +47,17 @@ absl::StatusOr BuildAttributesMap( auto integer = [&](mlir::IntegerAttr integer) { if (integer.getType().isUnsignedInteger()) { switch (integer.getType().getIntOrFloatBitWidth()) { - case 1: - attributes[name] = static_cast(integer.getInt()); - return absl::OkStatus(); case 8: - attributes[name] = static_cast(integer.getInt()); + attributes[name] = static_cast(integer.getUInt()); return absl::OkStatus(); case 16: - attributes[name] = static_cast(integer.getInt()); + attributes[name] = static_cast(integer.getUInt()); return absl::OkStatus(); case 32: - attributes[name] = static_cast(integer.getInt()); + attributes[name] = static_cast(integer.getUInt()); return absl::OkStatus(); case 64: - attributes[name] = static_cast(integer.getInt()); + attributes[name] = static_cast(integer.getUInt()); return absl::OkStatus(); default: return absl::InvalidArgumentError(absl::StrCat( @@ -69,9 +66,6 @@ absl::StatusOr BuildAttributesMap( } } else { switch (integer.getType().getIntOrFloatBitWidth()) { - case 1: - attributes[name] = static_cast(integer.getInt()); - return absl::OkStatus(); case 8: attributes[name] = static_cast(integer.getInt()); return absl::OkStatus();