Mixed poisson distributed in work

Wells-Group · Jul 18, 2024 · b93d70d · b93d70d
1 parent 64f668a
commit b93d70d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/demo/mixed_poisson_dlrbnicsx_distributed/dlrbnicsx_mixed_poisson_distributed.py b/demo/mixed_poisson_dlrbnicsx_distributed/dlrbnicsx_mixed_poisson_distributed.py
@@ -349,6 +349,8 @@ def generate_training_set(samples=num_pod_samples_sigma):
 for i in range(len(fem_comm_list)):
     cpu_indices_sigma = np.arange(i, para_matrix_sigma.shape[0], len(fem_comm_list))
 
+print(f"cpu_indices_sigma: {cpu_indices_sigma}")
+
 if world_comm.rank == 0:
     nbytes_dofs_sigma = num_snapshots_sigma * num_dofs_sigma * itemsize
 else: