wip

graphcore-research · Oct 2, 2023 · a5f9004 · a5f9004
1 parent 49a2e49
commit a5f9004
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/tessellate_ipu/core/tile_interpreter_vertex_utils.py b/tessellate_ipu/core/tile_interpreter_vertex_utils.py
@@ -39,8 +39,6 @@ def make_ipu_vector1d_worker_offsets(
     Returns:
         (6,) number of data vectors per thread.
     """
-    print("SIZE:", size)
-
     def make_offsets_fn(sizes):
         sizes = [0] + sizes
         offsets = np.cumsum(np.array(sizes, wdtype), dtype=wdtype)

diff --git a/tessellate_ipu/linalg/tile_linalg_jacobi.py b/tessellate_ipu/linalg/tile_linalg_jacobi.py
@@ -208,12 +208,12 @@ def ipu_jacobi_eigh_iteration(all_AV_cols: Tuple[Array, ...], Atiles: Any, Vtile
             halfN=halfN,
         )
         # Jacobi eigenvectors update step.
-        # Vpcols, Vqcols = tile_map(  # type:ignore
-        #     jacobi_update_eigenvectors_p,
-        #     cs_Vtiles,
-        #     Vpcols,
-        #     Vqcols,
-        # )
+        Vpcols, Vqcols = tile_map(  # type:ignore
+            jacobi_update_eigenvectors_p,
+            cs_Vtiles,
+            Vpcols,
+            Vqcols,
+        )
 
         # Barrier, to make we sync. both set of tiles A and V
         Apcols, Aqcols, Vpcols, Vqcols = tile_data_barrier(Apcols, Aqcols, Vpcols, Vqcols)