from st.expr import Index, ConstRef
from st.grid import Grid
# Declare indices
i = Index(0)
j = Index(1)
k = Index(2)
# Declare grid
input = Grid("in", 3)
output = Grid("out", 3)
c = []
for cc in range(125):
c.append(ConstRef('c[{}]'.format(cc)))
calc = c[0] * input(i, j, k) + c[1] * input(i - 3, j, k) + c[2] * input(
i - 2, j, k
) + c[3] * input(i - 1, j, k) + c[4] * input(i + 1, j, k) + c[5] * input(
i + 2, j, k) + c[6] * input(i + 3, j, k) + c[7] * input(
i, j - 3, k) + c[8] * input(i, j - 2, k) + c[9] * input(
i, j - 1, k) + c[10] * input(i, j + 1, k) + c[11] * input(
i, j + 2, k) + c[12] * input(i, j + 3, k) + c[13] * input(
i, j, k - 3) + c[14] * input(i, j, k - 2) + c[15] * input(
i, j, k - 1) + c[16] * input(
i, j, k + 1) + c[17] * input(
i, j, k + 2) + c[18] * input(i, j, k + 3)
output(i, j, k).assign(calc)
STENCIL = [output]
def genStoreLoc(self, grid: Grid, shift, offset, rel, dimrels):
from st.grid import GridRef
from st.expr import Index
dims = grid.dims
return self.gen_rhs(GridRef(grid, [Index(i) for i in range(dims)]),
shift, offset, rel, dimrels)
from st.expr import Index, ConstRef
from st.grid import Grid
# Declare indices
i = Index(0)
j = Index(1)
# Declare grid
input = Grid("in", 2)
output = Grid("out", 2)
c = []
for cc in range(25):
c.append(ConstRef('c[{}]'.format(cc)))
calc = c[0] * input(i - 2, j - 2) + c[1] * input(i - 1, j - 2) + c[2] * input(
i, j - 2) + c[3] * input(i + 1, j - 2) + c[4] * input(i + 2, j - 2) + c[
5] * input(i - 2, j - 1) + c[6] * input(i - 1, j - 1) + c[7] * input(
i, j - 1) + c[8] * input(i + 1, j - 1) + c[9] * input(
i + 2, j - 1) + c[10] * input(i - 2, j) + c[11] * input(
i - 1, j) + c[12] * input(i, j) + c[13] * input(
i + 1, j) + c[14] * input(i + 2, j) + c[15] * input(
i - 2, j + 1) + c[16] * input(
i - 1, j +
1) + c[17] * input(i, j + 1) + c[18] * input(
i + 1, j + 1) + c[19] * input(
i + 2, j + 1) + c[20] * input(
i - 2, j + 2) + c[21] * input(
i - 1, j + 2) + c[22] * input(
i, j + 2) + c[23] * input(
i + 1,
j + 2) + c[24] * input(