oryx.core.interpreters.harvest.HarvestTrace

View source on GitHub

A HarvestTrace manages HarvestTracer objects.

oryx.core.interpreters.harvest.HarvestTrace(
    main: "MainTrace",
    sublevel: "Sublevel"
) -> None

Since HarvestTracers are just wrappers around known values, HarvestTrace just passes these values through primitives, except in the case of sow and nest, which are specially handled by the active HarvestContext.

Default primitive logic lives in process_primitive, with special logic for sow in handle_sow.

Methods

default_process_primitive

View source

default_process_primitive(
    primitive, tracers, params
)

Handles primitives without custom harvest rules.

full_raise

full_raise(
    val
) -> "Tracer"

handle_sow

View source

handle_sow(
    name, *tracers, tag, mode, tree
)

instantiate_const

View source

instantiate_const(
    val
)

lift

View source

lift(
    val
)

post_process_call

View source

post_process_call(
    call_primitive, out_tracers, params
)

post_process_map

View source

post_process_map(
    call_primitive, out_tracers, params
)

process_call

View source

process_call(
    call_primitive, f, tracers, params
)

process_custom_jvp_call

View source

process_custom_jvp_call(
    primitive, fun, jvp, tracers
)

process_custom_vjp_call

View source

process_custom_vjp_call(
    primitive, fun, fwd, bwd, tracers, out_trees
)

process_higher_order_primitive

View source

process_higher_order_primitive(
    primitive, f, tracers, params, is_map
)

process_map

View source

process_map(
    call_primitive, f, tracers, params
)

process_primitive

View source

process_primitive(
    primitive, tracers, params
)

process_xmap

process_xmap(
    call_primitive, f, tracers, params
)

pure

View source

pure(
    val
)

sublift

View source

sublift(
    tracer
)

Class Variables
level	Instance of member_descriptor
main	Instance of member_descriptor
sublevel	Instance of member_descriptor