Accera v1.2 Reference

accera.Nest.iteration_logic(logic)

Adds an iteration logic function to a Nest.

Arguments

argument	description	type/default
logic		Python function that represents the logic to run in the innermost loop of the nest.

Examples

The preferred syntax uses Python decorators, as follows:

import accera as acc

A = acc.Array(role=acc.role.INPUT, shape=(16, 64))
B = acc.Array(role=acc.role.INPUT, shape=(64, 32))
C = acc.Array(role=acc.role.INPUT_OUTPUT, shape=(16, 32))

nest = acc.Nest(shape=(16, 32, 64))
i, j, k = nest.get_indices()

@nest.iteration_logic
def _():
    C[i,j] += A[i,k] * B[k,j]

The alternative syntax avoids decorators and instead defines the logic in a function:

import accera as acc

A = acc.Array(role=acc.role.INPUT, shape=(16, 64))
B = acc.Array(role=acc.role.INPUT, shape=(64, 32))
C = acc.Array(role=acc.role.INPUT_OUTPUT, shape=(16, 32))

nest = acc.Nest(shape=(16, 32, 64))
i, j, k = nest.get_indices()

def logic_fn():
    C[i, j] += A[i, k] * B[k, j]

nest.iteration_logic(logic_fn)

---

Last update: 2023-04-17