diff --git a/sumpy/e2e.py b/sumpy/e2e.py
index 12e80b5f5d79781ab7ff78bcad8954e3e6ef3271..2d54361102b7c9d3741aceea2c56f3da0fff546c 100644
--- a/sumpy/e2e.py
+++ b/sumpy/e2e.py
@@ -209,9 +209,9 @@ class E2EFromCSR(E2EBase):
                         np.int32),
                     lp.ValueArg("nsrc_level_boxes,ntgt_level_boxes",
                         np.int32),
-                    lp.GlobalArg("src_expansions", None,
+                    lp.GlobalArg("src_expansions", None, dim_tags="f,f",
                         shape=("nsrc_level_boxes", ncoeff_src), offset=lp.auto),
-                    lp.GlobalArg("tgt_expansions", None,
+                    lp.GlobalArg("tgt_expansions", None, dim_tags="f,f",
                         shape=("ntgt_level_boxes", ncoeff_tgt), offset=lp.auto),
                     "..."
                 ] + gather_loopy_arguments([self.src_expansion, self.tgt_expansion]),
@@ -328,9 +328,9 @@ class E2EFromChildren(E2EBase):
                     lp.ValueArg("src_rscale,tgt_rscale", None),
                     lp.GlobalArg("box_child_ids", None,
                         shape="nchildren, aligned_nboxes"),
-                    lp.GlobalArg("tgt_expansions", None,
+                    lp.GlobalArg("tgt_expansions", None, dim_tags="f,f",
                         shape=("ntgt_level_boxes", ncoeffs), offset=lp.auto),
-                    lp.GlobalArg("src_expansions", None,
+                    lp.GlobalArg("src_expansions", None, dim_tags="f,f",
                         shape=("nsrc_level_boxes", ncoeffs), offset=lp.auto),
                     lp.ValueArg("src_base_ibox,tgt_base_ibox", np.int32),
                     lp.ValueArg("ntgt_level_boxes,nsrc_level_boxes", np.int32),
@@ -436,9 +436,9 @@ class E2EFromParent(E2EBase):
                     lp.ValueArg("tgt_base_ibox,src_base_ibox", np.int32),
                     lp.ValueArg("ntgt_level_boxes,nsrc_level_boxes", np.int32),
                     lp.GlobalArg("box_parent_ids", None, shape="nboxes"),
-                    lp.GlobalArg("tgt_expansions", None,
+                    lp.GlobalArg("tgt_expansions", None, dim_tags="f,f",
                         shape=("ntgt_level_boxes", ncoeffs), offset=lp.auto),
-                    lp.GlobalArg("src_expansions", None,
+                    lp.GlobalArg("src_expansions", None, dim_tags="f,f",
                         shape=("nsrc_level_boxes", ncoeffs), offset=lp.auto),
                     "..."
                 ] + gather_loopy_arguments([self.src_expansion, self.tgt_expansion]),
diff --git a/sumpy/e2p.py b/sumpy/e2p.py
index d0d05cbf8c45e8d9f1c4552f784950bdfcab4bfb..2e9211b790b1bd38cf1a52aa5b0617e3299ac4f6 100644
--- a/sumpy/e2p.py
+++ b/sumpy/e2p.py
@@ -178,7 +178,7 @@ class E2PFromSingleBox(E2PBase):
                     lp.ValueArg("rscale", None),
                     lp.GlobalArg("result", None, shape="nresults, ntargets",
                         dim_tags="sep,C"),
-                    lp.GlobalArg("src_expansions", None,
+                    lp.GlobalArg("src_expansions", None, dim_tags="f,f",
                         shape=("nsrc_level_boxes", ncoeffs), offset=lp.auto),
                     lp.ValueArg("nsrc_level_boxes,naligned_boxes", np.int32),
                     lp.ValueArg("src_base_ibox", np.int32),
@@ -280,7 +280,7 @@ class E2PFromCSR(E2PBase):
                     lp.GlobalArg("box_target_starts,box_target_counts_nonchild",
                         None, shape=None),
                     lp.GlobalArg("centers", None, shape="dim, aligned_nboxes"),
-                    lp.GlobalArg("src_expansions", None,
+                    lp.GlobalArg("src_expansions", None, dim_tags="f,f",
                         shape=("nsrc_level_boxes", ncoeffs), offset=lp.auto),
                     lp.ValueArg("src_base_ibox", np.int32),
                     lp.ValueArg("nsrc_level_boxes,aligned_nboxes", np.int32),
diff --git a/sumpy/p2e.py b/sumpy/p2e.py
index daa7d93dc5b393d358424031500a4915e2c2dd9f..b9bc5822ba0b5f85ef3d1939986301604c30a1e9 100644
--- a/sumpy/p2e.py
+++ b/sumpy/p2e.py
@@ -150,7 +150,7 @@ class P2EFromSingleBox(P2EBase):
                         None, shape=None),
                     lp.GlobalArg("centers", None, shape="dim, aligned_nboxes"),
                     lp.ValueArg("rscale", None),
-                    lp.GlobalArg("tgt_expansions", None,
+                    lp.GlobalArg("tgt_expansions", None, dim_tags="f,f",
                         shape=("nboxes", ncoeffs), offset=lp.auto),
                     lp.ValueArg("nboxes,aligned_nboxes,tgt_base_ibox", np.int32),
                     lp.ValueArg("nsources", np.int32),
@@ -217,7 +217,7 @@ class P2EFromCSR(P2EBase):
                     lp.GlobalArg("box_source_starts,box_source_counts_nonchild",
                         None, shape=None),
                     lp.GlobalArg("centers", None, shape="dim, naligned_boxes"),
-                    lp.GlobalArg("tgt_expansions", None,
+                    lp.GlobalArg("tgt_expansions", None, dim_tags="f,f",
                         shape=("ntgt_level_boxes", ncoeffs), offset=lp.auto),
                     lp.ValueArg("naligned_boxes,ntgt_level_boxes,tgt_base_ibox",
                         np.int32),