void Set(T new_value, unsigned size = sizeof(T)) {
ASSERT(size <= kSizeInBytes);
ASSERT(size <= sizeof(new_value));
+ STATIC_ASSERT(kXRegSize == kDRegSize);
+ STATIC_ASSERT(kWRegSize == kSRegSize);
// All AArch64 registers are zero-extending; Writing a W register clears the
// top bits of the corresponding X register.
- memset(value_, 0, kSizeInBytes);
- memcpy(value_, &new_value, size);
+ if (size == kXRegSize) {
+ memcpy(value_, &new_value, kXRegSize);
+ } else if (size == kWRegSize) {
+ memset(value_, 0, kSizeInBytes);
+ memcpy(value_, &new_value, kWRegSize);
+ } else {
+ UNREACHABLE();
+ }
}
// Copy 'size' bytes of the register to the result, and zero-extend to fill
T Get(unsigned size = sizeof(T)) const {
ASSERT(size <= kSizeInBytes);
T result;
- memset(&result, 0, sizeof(result));
- memcpy(&result, value_, size);
+ if (size == kXRegSize) {
+ memcpy(&result, value_, kXRegSize);
+ } else if (size == kWRegSize) {
+ memset(&result, 0, sizeof(result));
+ memcpy(&result, value_, kWRegSize);
+ } else {
+ UNREACHABLE();
+ }
return result;
}