From 9eb29be9bb80afd10434c0835d4fc3be78cab1d5 Mon Sep 17 00:00:00 2001 From: samkellerhals Date: Wed, 11 Dec 2024 10:43:38 +0100 Subject: [PATCH] Add ConnectivityFields to handler --- src/gt4py/next/program_processors/runners/gtfn.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/gt4py/next/program_processors/runners/gtfn.py b/src/gt4py/next/program_processors/runners/gtfn.py index 1e7f530fd6..a2e1b62569 100644 --- a/src/gt4py/next/program_processors/runners/gtfn.py +++ b/src/gt4py/next/program_processors/runners/gtfn.py @@ -20,7 +20,7 @@ from gt4py.eve import utils from gt4py.eve.utils import content_hash from gt4py.next import NeighborTableOffsetProvider, backend, common, config -from gt4py.next.embedded.nd_array_field import NumPyArrayField +from gt4py.next.embedded import nd_array_field from gt4py.next.iterator import ir as itir from gt4py.next.otf import arguments from gt4py.next.otf import recipes, stages, workflow @@ -115,6 +115,8 @@ def handle_invalid_type( type_handlers_connectivity_args = { NeighborTableOffsetProvider: handle_connectivity, + nd_array_field.NumPyArrayConnectivityField: handle_connectivity, + nd_array_field.CuPyArrayConnectivityField: handle_connectivity, common.Dimension: handle_dimension, }