Port to actx.np.zeros

Author: alexfikl

sumpy/fmm.py

@@ -426,7 +426,7 @@ def multipole_expansion_zeros(self, actx: ArrayContext) -> Array:
         capable of holding one multipole or local expansion for every
         box in the tree.
         """
-        return actx.zeros(
+        return actx.np.zeros(
                 self.multipole_expansions_level_starts()[-1],
                 dtype=self.dtype)
 
@@ -435,7 +435,7 @@ def local_expansion_zeros(self, actx) -> Array:
         capable of holding one multipole or local expansion for every
         box in the tree.
         """
-        return actx.zeros(
+        return actx.np.zeros(
                 self.local_expansions_level_starts()[-1],
                 dtype=self.dtype)
 
@@ -452,7 +452,7 @@ def m2l_translation_classes_dependent_data_zeros(
             translation_class_start, translation_class_stop = (
                 level_start_box_nrs[level:level + 2])
 
-            exprs_level = actx.zeros(
+            exprs_level = actx.np.zeros(
                 expn_stop - expn_start,
                 dtype=self.preprocessed_mpole_dtype
                 ).reshape(translation_class_stop - translation_class_start, -1)
@@ -504,7 +504,7 @@ def m2l_preproc_mpole_expansion_zeros(
             expn_start, expn_stop = level_starts[level:level+2]
             box_start, box_stop = self.tree_level_start_box_nrs[level:level+2]
 
-            exprs_level = actx.zeros(
+            exprs_level = actx.np.zeros(
                 expn_stop - expn_start,
                 dtype=self.preprocessed_mpole_dtype,
                 ).reshape(box_stop - box_start, -1)
@@ -530,7 +530,7 @@ def output_zeros(self,
         the potential, or arbitrary other per-target output data.
         """
         return obj_array.new_1d([
-                actx.zeros(self.tree.ntargets, dtype=self.dtype)
+                actx.np.zeros(self.tree.ntargets, dtype=self.dtype)
                 for k in self.tree_indep.target_kernels])
 
     def reorder_sources(self, source_array):

sumpy/toys.py

@@ -403,7 +403,7 @@ def _m2l(actx: ArrayContext,
         expn_size = translation_classes_kwargs["m2l_expn_size"]
 
         # Preprocess the mpole expansion
-        preprocessed_src_expansions = actx.zeros((1, expn_size), dtype=np.complex128)
+        preprocessed_src_expansions = actx.np.zeros((1, expn_size), dtype=np.complex128)
         preprocess_kernel(
                 actx,
                 src_expansions=coeffs,
@@ -430,7 +430,7 @@ def _m2l(actx: ArrayContext,
         dim = toy_ctx.kernel.dim
         m2l_translation_vectors = actx.from_numpy(dist.reshape(dim, 1))
         m2l_translation_classes_dependent_data = (
-            actx.zeros((1, expn_size), dtype=np.complex128))
+            actx.np.zeros((1, expn_size), dtype=np.complex128))
 
         data_kernel(
                 actx,
@@ -462,7 +462,7 @@ def _m2l(actx: ArrayContext,
 
         # Postprocess the local expansion
         local_before = actx.from_numpy(np.array([ret.coeffs]))
-        to_coeffs = actx.zeros((1, len(data_kernel.tgt_expansion)),
+        to_coeffs = actx.np.zeros((1, len(data_kernel.tgt_expansion)),
                                dtype=coeffs.dtype)
 
         if toy_ctx.use_fft: