Added heat stencil and matmul assembly