Ops.capi.cpp.inc

Go to the documentation of this file.

/*===- TableGen'erated file -------------------------------------*- C++ -*-===*\
|*                                                                            *|
|* Op C API Definitions                                                       *|
|*                                                                            *|
|* Automatically generated file, do not edit!                                 *|
|* From: Ops.td                                                               *|
|*                                                                            *|
\*===----------------------------------------------------------------------===*/
 
 
#include <limits>
 
using namespace mlir;
using namespace llvm;
 
bool llzkOperationIsA_Pod_NewPodOp(MlirOperation inp) {
  return llvm::isa<NewPodOp>(unwrap(inp));
}
 
intptr_t llzkPod_NewPodOpGetInitialValuesCount(MlirOperation op) {
  auto range = llvm::cast<NewPodOp>(unwrap(op)).getODSOperandIndexAndLength(0);
  return range.second;
}
 
MlirValue llzkPod_NewPodOpGetInitialValuesAt(MlirOperation op, intptr_t index) {
  auto range = llvm::cast<NewPodOp>(unwrap(op)).getODSOperandIndexAndLength(0);
  assert(index >= 0 && index < range.second && "variadic operand index out of range");
  assert(
      static_cast<uintptr_t>(range.first) <= static_cast<uintptr_t>(std::numeric_limits<intptr_t>::max()) &&
      "operand index exceeds intptr_t range"
  );
  return mlirOperationGetOperand(op, static_cast<intptr_t>(range.first) + index);
}
 
void llzkPod_NewPodOpSetInitialValues(MlirOperation op, intptr_t count, MlirValue const *values) {
  if (count < 0)
    return;
  ::llvm::SmallVector<::mlir::Value> vals;
  vals.reserve(static_cast<size_t>(count));
  for (intptr_t i = 0; i < count; ++i)
    vals.push_back(unwrap(values[i]));
  ::llvm::cast<NewPodOp>(unwrap(op)).getInitialValuesMutable().assign(vals);
}
 
intptr_t llzkPod_NewPodOpGetMapOperandsCount(MlirOperation op) {
  auto range = llvm::cast<NewPodOp>(unwrap(op)).getODSOperandIndexAndLength(1);
  return range.second;
}
 
MlirValue llzkPod_NewPodOpGetMapOperandsAt(MlirOperation op, intptr_t index) {
  auto range = llvm::cast<NewPodOp>(unwrap(op)).getODSOperandIndexAndLength(1);
  assert(index >= 0 && index < range.second && "variadic operand index out of range");
  assert(
      static_cast<uintptr_t>(range.first) <= static_cast<uintptr_t>(std::numeric_limits<intptr_t>::max()) &&
      "operand index exceeds intptr_t range"
  );
  return mlirOperationGetOperand(op, static_cast<intptr_t>(range.first) + index);
}
 
void llzkPod_NewPodOpSetMapOperands(MlirOperation op, intptr_t groupCount, MlirValueRange const *groups) {
  if (groupCount < 0)
    return;
 
  ::llvm::SmallVector<::mlir::Value> vals;
  for (intptr_t g = 0; g < groupCount; ++g) {
    assert(groups[g].size >= 0 && "group size must be non-negative");
    for (intptr_t i = 0; i < groups[g].size; ++i) {
      vals.push_back(unwrap(groups[g].values[i]));
    }
  }
  ::llvm::cast<NewPodOp>(unwrap(op)).getMapOperandsMutable().join().assign(vals);
 
  ::llvm::SmallVector<int32_t> newGroupSizes;
  newGroupSizes.reserve(static_cast<size_t>(groupCount));
  for (intptr_t g = 0; g < groupCount; ++g) {
    assert(
        groups[g].size <= static_cast<intptr_t>(std::numeric_limits<int32_t>::max()) &&
        "group size exceeds int32_t range"
    );
    newGroupSizes.push_back(static_cast<int32_t>(groups[g].size));
  }
  MlirContext ctx = mlirOperationGetContext(op);
  assert(
      newGroupSizes.size() <= static_cast<size_t>(std::numeric_limits<intptr_t>::max()) &&
      "group count exceeds intptr_t range"
  );
  mlirOperationSetAttributeByName(
      op, mlirStringRefCreateFromCString("mapOpGroupSizes"),
      mlirDenseI32ArrayGet(ctx, static_cast<intptr_t>(newGroupSizes.size()), newGroupSizes.data())
  );
}
 
MlirAttribute llzkPod_NewPodOpGetInitializedRecords(MlirOperation op) {
  return mlirOperationGetAttributeByName(op, mlirStringRefCreateFromCString("initializedRecords"));
}
 
void llzkPod_NewPodOpSetInitializedRecords(MlirOperation op, MlirAttribute attr) {
  mlirOperationSetAttributeByName(op, mlirStringRefCreateFromCString("initializedRecords"), attr);
}
 
MlirAttribute llzkPod_NewPodOpGetNumDimsPerMap(MlirOperation op) {
  return mlirOperationGetAttributeByName(op, mlirStringRefCreateFromCString("numDimsPerMap"));
}
 
void llzkPod_NewPodOpSetNumDimsPerMap(MlirOperation op, MlirAttribute attr) {
  mlirOperationSetAttributeByName(op, mlirStringRefCreateFromCString("numDimsPerMap"), attr);
}
 
MlirAttribute llzkPod_NewPodOpGetMapOpGroupSizes(MlirOperation op) {
  return mlirOperationGetAttributeByName(op, mlirStringRefCreateFromCString("mapOpGroupSizes"));
}
 
void llzkPod_NewPodOpSetMapOpGroupSizes(MlirOperation op, MlirAttribute attr) {
  mlirOperationSetAttributeByName(op, mlirStringRefCreateFromCString("mapOpGroupSizes"), attr);
}
 
MlirValue llzkPod_NewPodOpGetResult(MlirOperation op) {
  return mlirOperationGetResult(op, 0);
}
 
MlirOperation llzkPod_ReadPodOpBuild(MlirOpBuilder builder, MlirLocation location, MlirType resultType, MlirValue pod_ref, MlirAttribute record_name) {
  MlirOperationState state = mlirOperationStateGet(mlirStringRefCreateFromCString("pod.read"), location);
  mlirOperationStateAddResults(&state, 1, &resultType);
  mlirOperationStateAddOperands(&state, 1, &pod_ref);
  MlirContext ctx = mlirOpBuilderGetContext(builder);
  llvm::SmallVector<MlirNamedAttribute, 1> attributes;
  if (!mlirAttributeIsNull(record_name)) {
    attributes.push_back(mlirNamedAttributeGet(mlirIdentifierGet(ctx, mlirStringRefCreateFromCString("record_name")), record_name));
  }
  mlirOperationStateAddAttributes(&state, attributes.size(), attributes.data());
 
  return mlirOpBuilderInsert(builder, mlirOperationCreate(&state));
}
 
bool llzkOperationIsA_Pod_ReadPodOp(MlirOperation inp) {
  return llvm::isa<ReadPodOp>(unwrap(inp));
}
 
MlirValue llzkPod_ReadPodOpGetPodRef(MlirOperation op) {
  auto range = llvm::cast<ReadPodOp>(unwrap(op)).getODSOperandIndexAndLength(0);
  assert(range.second == 1 && "expected fixed operand segment size");
  assert(
      static_cast<uintptr_t>(range.first) <= static_cast<uintptr_t>(std::numeric_limits<intptr_t>::max()) &&
      "operand index exceeds intptr_t range"
  );
  return mlirOperationGetOperand(op, static_cast<intptr_t>(range.first));
}
 
void llzkPod_ReadPodOpSetPodRef(MlirOperation op, MlirValue value) {
  auto range = llvm::cast<ReadPodOp>(unwrap(op)).getODSOperandIndexAndLength(0);
  assert(range.second == 1 && "expected fixed operand segment size");
  assert(
      static_cast<uintptr_t>(range.first) <= static_cast<uintptr_t>(std::numeric_limits<intptr_t>::max()) &&
      "operand index exceeds intptr_t range"
  );
  mlirOperationSetOperand(op, static_cast<intptr_t>(range.first), value);
}
 
MlirAttribute llzkPod_ReadPodOpGetRecordName(MlirOperation op) {
  return mlirOperationGetAttributeByName(op, mlirStringRefCreateFromCString("record_name"));
}
 
void llzkPod_ReadPodOpSetRecordName(MlirOperation op, MlirAttribute attr) {
  mlirOperationSetAttributeByName(op, mlirStringRefCreateFromCString("record_name"), attr);
}
 
MlirValue llzkPod_ReadPodOpGetResult(MlirOperation op) {
  return mlirOperationGetResult(op, 0);
}
 
MlirType llzkPod_ReadPodOpGetPodRefType(MlirOperation inp) {
  return wrap(llvm::cast<ReadPodOp>(unwrap(inp)).getPodRefType());
}
 
MlirOperation llzkPod_WritePodOpBuild(MlirOpBuilder builder, MlirLocation location, MlirValue pod_ref, MlirValue value, MlirAttribute record_name) {
  MlirOperationState state = mlirOperationStateGet(mlirStringRefCreateFromCString("pod.write"), location);
  mlirOperationStateAddOperands(&state, 1, &pod_ref);
  mlirOperationStateAddOperands(&state, 1, &value);
  MlirContext ctx = mlirOpBuilderGetContext(builder);
  llvm::SmallVector<MlirNamedAttribute, 1> attributes;
  if (!mlirAttributeIsNull(record_name)) {
    attributes.push_back(mlirNamedAttributeGet(mlirIdentifierGet(ctx, mlirStringRefCreateFromCString("record_name")), record_name));
  }
  mlirOperationStateAddAttributes(&state, attributes.size(), attributes.data());
 
  return mlirOpBuilderInsert(builder, mlirOperationCreate(&state));
}
 
bool llzkOperationIsA_Pod_WritePodOp(MlirOperation inp) {
  return llvm::isa<WritePodOp>(unwrap(inp));
}
 
MlirValue llzkPod_WritePodOpGetPodRef(MlirOperation op) {
  auto range = llvm::cast<WritePodOp>(unwrap(op)).getODSOperandIndexAndLength(0);
  assert(range.second == 1 && "expected fixed operand segment size");
  assert(
      static_cast<uintptr_t>(range.first) <= static_cast<uintptr_t>(std::numeric_limits<intptr_t>::max()) &&
      "operand index exceeds intptr_t range"
  );
  return mlirOperationGetOperand(op, static_cast<intptr_t>(range.first));
}
 
void llzkPod_WritePodOpSetPodRef(MlirOperation op, MlirValue value) {
  auto range = llvm::cast<WritePodOp>(unwrap(op)).getODSOperandIndexAndLength(0);
  assert(range.second == 1 && "expected fixed operand segment size");
  assert(
      static_cast<uintptr_t>(range.first) <= static_cast<uintptr_t>(std::numeric_limits<intptr_t>::max()) &&
      "operand index exceeds intptr_t range"
  );
  mlirOperationSetOperand(op, static_cast<intptr_t>(range.first), value);
}
 
MlirValue llzkPod_WritePodOpGetValue(MlirOperation op) {
  auto range = llvm::cast<WritePodOp>(unwrap(op)).getODSOperandIndexAndLength(1);
  assert(range.second == 1 && "expected fixed operand segment size");
  assert(
      static_cast<uintptr_t>(range.first) <= static_cast<uintptr_t>(std::numeric_limits<intptr_t>::max()) &&
      "operand index exceeds intptr_t range"
  );
  return mlirOperationGetOperand(op, static_cast<intptr_t>(range.first));
}
 
void llzkPod_WritePodOpSetValue(MlirOperation op, MlirValue value) {
  auto range = llvm::cast<WritePodOp>(unwrap(op)).getODSOperandIndexAndLength(1);
  assert(range.second == 1 && "expected fixed operand segment size");
  assert(
      static_cast<uintptr_t>(range.first) <= static_cast<uintptr_t>(std::numeric_limits<intptr_t>::max()) &&
      "operand index exceeds intptr_t range"
  );
  mlirOperationSetOperand(op, static_cast<intptr_t>(range.first), value);
}
 
MlirAttribute llzkPod_WritePodOpGetRecordName(MlirOperation op) {
  return mlirOperationGetAttributeByName(op, mlirStringRefCreateFromCString("record_name"));
}
 
void llzkPod_WritePodOpSetRecordName(MlirOperation op, MlirAttribute attr) {
  mlirOperationSetAttributeByName(op, mlirStringRefCreateFromCString("record_name"), attr);
}
 
MlirType llzkPod_WritePodOpGetPodRefType(MlirOperation inp) {
  return wrap(llvm::cast<WritePodOp>(unwrap(inp)).getPodRefType());
}