318b0dda7c
This patch adds support for atomic loads and stores. Specifically, it adds support for the following intrinsic calls: - `__atomic_load` and `__atomic_store`; - `__c11_atomic_load` and `__c11_atomic_store`.
126 lines
3.9 KiB
C++
126 lines
3.9 KiB
C++
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This class provides a simple wrapper for a pair of a pointer and an
|
|
// alignment.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef CLANG_LIB_CIR_ADDRESS_H
|
|
#define CLANG_LIB_CIR_ADDRESS_H
|
|
|
|
#include "mlir/IR/Value.h"
|
|
#include "clang/AST/CharUnits.h"
|
|
#include "clang/CIR/Dialect/IR/CIRTypes.h"
|
|
#include "llvm/ADT/PointerIntPair.h"
|
|
|
|
namespace clang::CIRGen {
|
|
|
|
// Forward declaration to avoid a circular dependency
|
|
class CIRGenBuilderTy;
|
|
|
|
class Address {
|
|
|
|
// The boolean flag indicates whether the pointer is known to be non-null.
|
|
llvm::PointerIntPair<mlir::Value, 1, bool> pointerAndKnownNonNull;
|
|
|
|
/// The expected CIR type of the pointer. Carrying accurate element type
|
|
/// information in Address makes it more convenient to work with Address
|
|
/// values and allows frontend assertions to catch simple mistakes.
|
|
mlir::Type elementType;
|
|
|
|
clang::CharUnits alignment;
|
|
|
|
protected:
|
|
Address(std::nullptr_t) : elementType(nullptr) {}
|
|
|
|
public:
|
|
Address(mlir::Value pointer, mlir::Type elementType,
|
|
clang::CharUnits alignment)
|
|
: pointerAndKnownNonNull(pointer, false), elementType(elementType),
|
|
alignment(alignment) {
|
|
assert(mlir::isa<cir::PointerType>(pointer.getType()) &&
|
|
"Expected cir.ptr type");
|
|
|
|
assert(pointer && "Pointer cannot be null");
|
|
assert(elementType && "Element type cannot be null");
|
|
assert(!alignment.isZero() && "Alignment cannot be zero");
|
|
|
|
assert(mlir::cast<cir::PointerType>(pointer.getType()).getPointee() ==
|
|
elementType);
|
|
}
|
|
|
|
Address(mlir::Value pointer, clang::CharUnits alignment)
|
|
: Address(pointer,
|
|
mlir::cast<cir::PointerType>(pointer.getType()).getPointee(),
|
|
alignment) {
|
|
assert((!alignment.isZero() || pointer == nullptr) &&
|
|
"creating valid address with invalid alignment");
|
|
}
|
|
|
|
static Address invalid() { return Address(nullptr); }
|
|
bool isValid() const {
|
|
return pointerAndKnownNonNull.getPointer() != nullptr;
|
|
}
|
|
|
|
/// Return address with different pointer, but same element type and
|
|
/// alignment.
|
|
Address withPointer(mlir::Value newPtr) const {
|
|
return Address(newPtr, getElementType(), getAlignment());
|
|
}
|
|
|
|
/// Return address with different element type, a bitcast pointer, and
|
|
/// the same alignment.
|
|
Address withElementType(CIRGenBuilderTy &builder, mlir::Type ElemTy) const;
|
|
|
|
mlir::Value getPointer() const {
|
|
assert(isValid());
|
|
return pointerAndKnownNonNull.getPointer();
|
|
}
|
|
|
|
mlir::Value getBasePointer() const {
|
|
// TODO(cir): Remove the version above when we catchup with OG codegen on
|
|
// ptr auth.
|
|
assert(isValid() && "pointer isn't valid");
|
|
return getPointer();
|
|
}
|
|
|
|
mlir::Type getType() const {
|
|
assert(mlir::cast<cir::PointerType>(
|
|
pointerAndKnownNonNull.getPointer().getType())
|
|
.getPointee() == elementType);
|
|
|
|
return mlir::cast<cir::PointerType>(getPointer().getType());
|
|
}
|
|
|
|
mlir::Type getElementType() const {
|
|
assert(isValid());
|
|
assert(mlir::cast<cir::PointerType>(
|
|
pointerAndKnownNonNull.getPointer().getType())
|
|
.getPointee() == elementType);
|
|
return elementType;
|
|
}
|
|
|
|
clang::CharUnits getAlignment() const { return alignment; }
|
|
|
|
/// Get the operation which defines this address.
|
|
mlir::Operation *getDefiningOp() const {
|
|
if (!isValid())
|
|
return nullptr;
|
|
return getPointer().getDefiningOp();
|
|
}
|
|
|
|
template <typename OpTy> OpTy getDefiningOp() const {
|
|
return mlir::dyn_cast_or_null<OpTy>(getDefiningOp());
|
|
}
|
|
};
|
|
|
|
} // namespace clang::CIRGen
|
|
|
|
#endif // CLANG_LIB_CIR_ADDRESS_H
|