Fix bug for the opt rect algo, close #42

cp2k · Oct 9, 2018 · a295923 · a295923
1 parent 1d923e0
commit a295923
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/mm/dbcsr_mm_dist_operations.F b/src/mm/dbcsr_mm_dist_operations.F
@@ -315,8 +315,13 @@ SUBROUTINE dbcsr_make_dists_dense(product_dist, left_rdist, right_rdist, &
       IF (.NOT. dbcsr_distribution_has_threads(product_dist)) &
          DBCSR_ABORT("Product distribution must have threads.")
       tdist => array_data(dbcsr_distribution_thread_dist(product_dist))
-      old_m_dist = product_dist%d%row_dist_block
-      old_n_dist = product_dist%d%col_dist_block
+      IF (local_mult) THEN
+         old_m_dist = left_rdist%i%main%d%row_dist_block
+         old_n_dist = right_rdist%i%main%d%col_dist_block
+      ELSE
+         old_m_dist = product_dist%d%row_dist_block
+         old_n_dist = product_dist%d%col_dist_block
+      ENDIF
       old_k_vdist = right_rdist%i%vrow_dist
       m_nbins = dbcsr_mp_nprows(product_dist%d%mp_env)
       n_nbins = dbcsr_mp_npcols(product_dist%d%mp_env)