EnzymeMLIR.cpp File Reference

#include "EnzymeMLIR.h"
#include "mlir/CAPI/IR.h"
#include "Dialect/Dialect.h"
#include "Dialect/Impulse/Impulse.h"
#include "Dialect/Ops.h"

Include dependency graph for EnzymeMLIR.cpp:

Go to the source code of this file.

Functions
MlirAttribute	enzymeRngDistributionAttrGet (MlirContext ctx, EnzymeRngDistribution dist)
MlirAttribute	enzymeSupportAttrGet (MlirContext ctx, EnzymeSupportKind kind, bool hasLowerBound, double lowerBound, bool hasUpperBound, double upperBound)
MlirAttribute	enzymeHMCConfigAttrGet (MlirContext ctx, double trajectoryLength, bool adaptStepSize, bool adaptMassMatrix)
MlirAttribute	enzymeNUTSConfigAttrGet (MlirContext ctx, int64_t maxTreeDepth, bool hasMaxDeltaEnergy, double maxDeltaEnergy, bool adaptStepSize, bool adaptMassMatrix)
MlirAttribute	enzymeSymbolAttrGet (MlirContext ctx, uint64_t ptr)

Function Documentation

enzymeHMCConfigAttrGet()

MlirAttribute enzymeHMCConfigAttrGet	(	MlirContext	ctx,
		double	trajectoryLength,
		bool	adaptStepSize,
		bool	adaptMassMatrix )

Definition at line 67 of file EnzymeMLIR.cpp.

enzymeNUTSConfigAttrGet()

MlirAttribute enzymeNUTSConfigAttrGet	(	MlirContext	ctx,
		int64_t	maxTreeDepth,
		bool	hasMaxDeltaEnergy,
		double	maxDeltaEnergy,
		bool	adaptStepSize,
		bool	adaptMassMatrix )

Definition at line 77 of file EnzymeMLIR.cpp.

enzymeRngDistributionAttrGet()

MlirAttribute enzymeRngDistributionAttrGet	(	MlirContext	ctx,
		EnzymeRngDistribution	dist )

Definition at line 9 of file EnzymeMLIR.cpp.

References EnzymeRngDistribution_MultiNormal, EnzymeRngDistribution_Normal, and EnzymeRngDistribution_Uniform.

enzymeSupportAttrGet()

MlirAttribute enzymeSupportAttrGet	(	MlirContext	ctx,
		EnzymeSupportKind	kind,
		bool	hasLowerBound,
		double	lowerBound,
		bool	hasUpperBound,
		double	upperBound )

Definition at line 26 of file EnzymeMLIR.cpp.

References EnzymeSupportKind_GreaterThan, EnzymeSupportKind_Interval, EnzymeSupportKind_LessThan, EnzymeSupportKind_Positive, EnzymeSupportKind_Real, and EnzymeSupportKind_UnitInterval.

enzymeSymbolAttrGet()

MlirAttribute enzymeSymbolAttrGet	(	MlirContext	ctx,
		uint64_t	ptr )

Definition at line 93 of file EnzymeMLIR.cpp.