IRAttributes.cpp File Reference

#include <cstdint>
#include <optional>
#include <string>
#include <string_view>
#include <utility>
#include "mlir-c/BuiltinAttributes.h"
#include "mlir-c/BuiltinTypes.h"
#include "mlir/Bindings/Python/IRAttributes.h"
#include "mlir/Bindings/Python/IRCore.h"
#include "mlir/Bindings/Python/Nanobind.h"
#include "mlir/Bindings/Python/NanobindAdaptors.h"
#include "mlir/Bindings/Python/NanobindUtils.h"
#include "llvm/ADT/ScopeExit.h"
#include "llvm/Support/raw_ostream.h"

Go to the source code of this file.

Classes
struct	mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::nb_format_descriptor< bool >
struct	mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::nb_format_descriptor< int8_t >
struct	mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::nb_format_descriptor< uint8_t >
struct	mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::nb_format_descriptor< int16_t >
struct	mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::nb_format_descriptor< uint16_t >
struct	mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::nb_format_descriptor< int32_t >
struct	mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::nb_format_descriptor< uint32_t >
struct	mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::nb_format_descriptor< int64_t >
struct	mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::nb_format_descriptor< uint64_t >
struct	mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::nb_format_descriptor< float >
struct	mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::nb_format_descriptor< double >

Namespaces
namespace	mlir
	Include the generated interface declarations.
namespace	mlir::python
namespace	mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN

Macros
#define	Py_IsFinalizing   _Py_IsFinalizing

Functions
nb::object	mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::denseArrayAttributeCaster (PyAttribute &pyAttribute)
nb::object	mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::denseIntOrFPElementsAttributeCaster (PyAttribute &pyAttribute)
nb::object	mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::integerOrBoolAttributeCaster (PyAttribute &pyAttribute)
nb::object	mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::symbolRefOrFlatSymbolRefAttributeCaster (PyAttribute &pyAttribute)
void	mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::populateIRAttributes (nb::module_ &m)

Variables
static const char	kDenseElementsAttrGetDocstring []
static const char	kDenseElementsAttrGetFromListDocstring []
static const char	kDenseResourceElementsAttrGetFromBufferDocstring []

Macro Definition Documentation

Py_IsFinalizing

#define Py_IsFinalizing   _Py_IsFinalizing

Definition at line 1032 of file IRAttributes.cpp.

Referenced by mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::PyDenseResourceElementsAttribute::getFromBuffer().

Variable Documentation

kDenseElementsAttrGetDocstring

const char kDenseElementsAttrGetDocstring[]

static

Definition at line 36 of file IRAttributes.cpp.

Referenced by mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::PyDenseElementsAttribute::bindDerived().

kDenseElementsAttrGetFromListDocstring

const char kDenseElementsAttrGetFromListDocstring[]

static

Initial value:

=
    R"(Gets a DenseElementsAttr from a Python list of attributes.
 
Note that it can be expensive to construct attributes individually.
For a large number of elements, consider using a Python buffer or array instead.
 
Args:
  attrs: A list of attributes.
  type: The desired shape and type of the resulting DenseElementsAttr.
    If not provided, the element type is determined based on the type
    of the 0th attribute and the shape is `[len(attrs)]`.
  context: Explicit context, if not from context manager.
 
Returns:
  DenseElementsAttr on success.
 
Raises:
  ValueError: If the type of the attributes does not match the type
    specified by `shaped_type`.
)"

Definition at line 78 of file IRAttributes.cpp.

Referenced by mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::PyDenseElementsAttribute::bindDerived().

kDenseResourceElementsAttrGetFromBufferDocstring

const char kDenseResourceElementsAttrGetFromBufferDocstring[]

static

Initial value:

=
    R"(Gets a DenseResourceElementsAttr from a Python buffer or array.
 
This function does minimal validation or massaging of the data, and it is
up to the caller to ensure that the buffer meets the characteristics
implied by the shape.
 
The backing buffer and any user objects will be retained for the lifetime
of the resource blob. This is typically bounded to the context but the
resource can have a shorter lifespan depending on how it is used in
subsequent processing.
 
Args:
  buffer: The array or buffer to convert.
  name: Name to provide to the resource (may be changed upon collision).
  type: The explicit ShapedType to construct the attribute with.
  context: Explicit context, if not from context manager.
 
Returns:
  DenseResourceElementsAttr on success.
 
Raises:
  ValueError: If the type of the buffer or array cannot be matched to an MLIR
    type or if the buffer does not meet expectations.
)"

Definition at line 99 of file IRAttributes.cpp.

Referenced by mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::PyDenseResourceElementsAttribute::bindDerived().