replace deprecated actx.empty with actx.zeros

Author: alexfikl

pytential/qbx/__init__.py

@@ -907,8 +907,8 @@ def _flat_centers(dofdesc, qbx_forced_limit):
 
             qbx_tgt_numberer = self.get_qbx_target_numberer(
                     tgt_to_qbx_center.dtype)
-            qbx_tgt_count = actx.empty((), np.int32)
-            qbx_tgt_numbers = actx.empty_like(tgt_to_qbx_center)
+            qbx_tgt_count = actx.zeros((), np.int32)
+            qbx_tgt_numbers = actx.np.zeros_like(tgt_to_qbx_center)
 
             qbx_tgt_numberer(
                     tgt_to_qbx_center, qbx_tgt_numbers, qbx_tgt_count,

pytential/qbx/geometry.py

@@ -481,7 +481,7 @@ def target_info(self):
 
         target_discr_starts.append(ntargets)
 
-        targets = actx.empty((self.ambient_dim, ntargets), self.coord_dtype)
+        targets = actx.zeros((self.ambient_dim, ntargets), self.coord_dtype)
         code_getter.copy_targets_kernel()(
                 actx.queue,
                 targets=targets[:, :self.ncenters],
@@ -510,9 +510,7 @@ def target_side_preferences(self):
         actx = self._setup_actx
         tgt_info = self.target_info()
 
-        target_side_preferences = actx.empty(tgt_info.ntargets, dtype=np.int8)
-        target_side_preferences[:self.ncenters] = 0
-
+        target_side_preferences = actx.zeros(tgt_info.ntargets, dtype=np.int8)
         for tdstart, (target_discr, qbx_side) in zip(
                 tgt_info.target_discr_starts,
                 self.target_discrs_and_qbx_sides):
@@ -549,7 +547,7 @@ def tree(self):
             target_radii = actx.zeros(target_info.ntargets, self.coord_dtype)
             target_radii[:self.ncenters] = self.flat_expansion_radii()
 
-        refine_weights = actx.empty(nparticles, dtype=np.int32)
+        refine_weights = actx.zeros(nparticles, dtype=np.int32)
 
         # Assign a weight of 1 to all sources, QBX centers, and conventional
         # (non-QBX) targets. Assign a weight of 0 to all targets that need
@@ -625,7 +623,7 @@ def qbx_center_to_target_box(self):
                         tree.box_id_dtype,
                         )
 
-        box_to_target_box = actx.empty(tree.nboxes, tree.box_id_dtype)
+        box_to_target_box = actx.zeros(tree.nboxes, tree.box_id_dtype)
         if self.debug:
             box_to_target_box.fill(-1)
 
@@ -634,15 +632,15 @@ def qbx_center_to_target_box(self):
                 )
 
         sorted_target_ids = self.tree().sorted_target_ids
-        user_target_from_tree_target = actx.empty_like(sorted_target_ids)
+        user_target_from_tree_target = actx.np.zeros_like(sorted_target_ids)
 
         user_target_from_tree_target[sorted_target_ids] = actx.from_numpy(
                 np.arange(
                     len(sorted_target_ids),
                     dtype=user_target_from_tree_target.dtype)
                 )
 
-        qbx_center_to_target_box = actx.empty(self.ncenters, tree.box_id_dtype)
+        qbx_center_to_target_box = actx.zeros(self.ncenters, tree.box_id_dtype)
 
         if self.debug:
             qbx_center_to_target_box.fill(-1)
@@ -677,7 +675,7 @@ def qbx_center_to_target_box_source_level(self, source_level):
         sep_smaller = traversal.from_sep_smaller_by_level[source_level]
         qbx_center_to_target_box = self.qbx_center_to_target_box()
 
-        target_box_to_target_box_source_level = actx.empty(
+        target_box_to_target_box_source_level = actx.zeros(
             len(traversal.target_boxes), dtype=traversal.tree.box_id_dtype)
         target_box_to_target_box_source_level.fill(-1)
         target_box_to_target_box_source_level[sep_smaller.nonempty_indices] = (
@@ -709,7 +707,7 @@ def global_qbx_flags(self):
         """
         actx = self._setup_actx
 
-        result = actx.empty(self.ncenters, np.int8)
+        result = actx.zeros(self.ncenters, np.int8)
         result.fill(1)
 
         return actx.freeze(result)
@@ -793,7 +791,7 @@ def user_target_to_center(self):
                     self.target_association_tolerance),
                 debug=self.debug)
 
-        result = actx.empty(
+        result = actx.zeros(
                 target_info.ntargets, tgt_assoc_result.target_to_center.dtype
                 )
         result[:self.ncenters].fill(target_state.NO_QBX_NEEDED)
@@ -812,12 +810,12 @@ def center_to_tree_targets(self):
         actx = self._setup_actx
         user_ttc = self.user_target_to_center()
 
-        tree_ttc = actx.empty_like(user_ttc)
+        tree_ttc = actx.np.zeros_like(user_ttc)
         tree_ttc[self.tree().sorted_target_ids] = user_ttc
 
-        filtered_tree_ttc = actx.empty(tree_ttc.shape, dtype=tree_ttc.dtype)
-        filtered_target_ids = actx.empty(tree_ttc.shape, dtype=tree_ttc.dtype)
-        count = actx.empty(1, dtype=tree_ttc.dtype)
+        filtered_tree_ttc = actx.zeros(tree_ttc.shape, dtype=tree_ttc.dtype)
+        filtered_target_ids = actx.zeros(tree_ttc.shape, dtype=tree_ttc.dtype)
+        count = actx.zeros(1, dtype=tree_ttc.dtype)
 
         self.code_getter.filter_center_and_target_ids(tree_ttc.dtype)(
                 tree_ttc, filtered_tree_ttc, filtered_target_ids, count,

pytential/qbx/target_assoc.py

@@ -658,7 +658,7 @@ def find_centers(self, places, dofdesc,
         wait_for = [evt]
 
         def make_target_field(fill_val, dtype=tree.coord_dtype):
-            arr = actx.empty(tree.nqbxtargets, dtype)
+            arr = actx.zeros(tree.nqbxtargets, dtype)
             arr.fill(fill_val)
             wait_for.extend(arr.events)
             return arr
@@ -790,7 +790,7 @@ def make_target_flags(self, target_discrs_and_qbx_sides):
         actx = self.array_context
 
         ntargets = sum(discr.ndofs for discr, _ in target_discrs_and_qbx_sides)
-        target_flags = actx.empty(ntargets, dtype=np.int32)
+        target_flags = actx.zeros(ntargets, dtype=np.int32)
 
         offset = 0
         for discr, flags in target_discrs_and_qbx_sides:
@@ -805,7 +805,7 @@ def make_target_flags(self, target_discrs_and_qbx_sides):
         return target_flags
 
     def make_default_target_association(self, ntargets):
-        target_to_center = self.array_context.empty(ntargets, dtype=np.int32)
+        target_to_center = self.array_context.zeros(ntargets, dtype=np.int32)
         target_to_center.fill(-1)
         target_to_center.finish()
 

pytential/qbx/utils.py

@@ -363,7 +363,7 @@ def _make_centers(discr):
     del box_to_class
 
     # Compute element => source relation
-    qbx_element_to_source_starts = actx.empty(nelements + 1, tree.particle_id_dtype)
+    qbx_element_to_source_starts = actx.zeros(nelements + 1, tree.particle_id_dtype)
     el_offset = 0
     node_nr_base = 0
     for group in density_discr.groups:

pytential/unregularized.py

@@ -443,7 +443,7 @@ def target_info(self):
         actx = self.array_context
         target_discr_starts.append(ntargets)
 
-        targets = actx.empty(
+        targets = actx.zeros(
                 (lpot_src.ambient_dim, ntargets),
                 self.coord_dtype)