EnzymeMLIR.h File Reference

#include <stdbool.h>
#include <stdint.h>
#include "mlir-c/IR.h"
#include "mlir-c/Support.h"

Include dependency graph for EnzymeMLIR.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Enumerations
enum	EnzymeRngDistribution { EnzymeRngDistribution_Uniform = 0 , EnzymeRngDistribution_Normal = 1 , EnzymeRngDistribution_MultiNormal = 2 }
enum	EnzymeSupportKind {   EnzymeSupportKind_Real = 0 , EnzymeSupportKind_Positive = 1 , EnzymeSupportKind_UnitInterval = 2 , EnzymeSupportKind_Interval = 3 ,   EnzymeSupportKind_GreaterThan = 4 , EnzymeSupportKind_LessThan = 5 }

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

Enumeration Type Documentation

EnzymeRngDistribution

enum EnzymeRngDistribution

Enumerator
EnzymeRngDistribution_Uniform
EnzymeRngDistribution_Normal
EnzymeRngDistribution_MultiNormal

Definition at line 18 of file EnzymeMLIR.h.

EnzymeSupportKind

enum EnzymeSupportKind

Enumerator
EnzymeSupportKind_Real
EnzymeSupportKind_Positive
EnzymeSupportKind_UnitInterval
EnzymeSupportKind_Interval
EnzymeSupportKind_GreaterThan
EnzymeSupportKind_LessThan

Definition at line 27 of file EnzymeMLIR.h.

Function Documentation

enzymeHMCConfigAttrGet()

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

Definition at line 67 of file EnzymeMLIR.cpp.

enzymeNUTSConfigAttrGet()

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

Definition at line 77 of file EnzymeMLIR.cpp.

enzymeRngDistributionAttrGet()

MLIR_CAPI_EXPORTED MlirAttribute enzymeRngDistributionAttrGet	(	MlirContext	ctx,
		EnzymeRngDistribution	dist )

Definition at line 9 of file EnzymeMLIR.cpp.

References EnzymeRngDistribution_MultiNormal, EnzymeRngDistribution_Normal, and EnzymeRngDistribution_Uniform.

enzymeSupportAttrGet()

MLIR_CAPI_EXPORTED 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()

MLIR_CAPI_EXPORTED MlirAttribute enzymeSymbolAttrGet	(	MlirContext	ctx,
		uint64_t	ptr )

Definition at line 93 of file EnzymeMLIR.cpp.