comp-chem/cp2k
1
0
Fork 0
mirror of https://github.com/cp2k/cp2k.git synced 2026-07-27 21:55:16 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

DBCSR-RMA-3D: fix a bug in the final reduction for GPU implementation

Browse source 
svn-origin-rev: 17244
This commit is contained in:
Alfio Lazzaro 2016-08-12 22:36:41 +00:00
parent f14b112b85
commit 7949710bbf
1 changed files with 12 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

17
src/dbcsr/mm/dbcsr_mm_3d.F
View file

@ -33,11 +33,11 @@ MODULE dbcsr_mm_3d
use_mpi_filtering
USE dbcsr_data_methods, ONLY: &
dbcsr_data_clear_pointer, dbcsr_data_ensure_size, dbcsr_data_get_size, &
dbcsr_data_get_type, dbcsr_data_host2dev, dbcsr_data_init, dbcsr_data_new, &
dbcsr_data_release, dbcsr_data_set_pointer, dbcsr_data_set_size_referenced, &
dbcsr_data_valid, dbcsr_get_data_p_c, dbcsr_get_data_p_d, dbcsr_get_data_p_s, &
dbcsr_get_data_p_z, dbcsr_scalar_are_equal, dbcsr_scalar_negative, dbcsr_scalar_one, &
dbcsr_type_1d_to_2d
dbcsr_data_get_type, dbcsr_data_hold, dbcsr_data_host2dev, dbcsr_data_init, &
dbcsr_data_new, dbcsr_data_release, dbcsr_data_set_pointer, &
dbcsr_data_set_size_referenced, dbcsr_data_valid, dbcsr_get_data_p_c, dbcsr_get_data_p_d, &
dbcsr_get_data_p_s, dbcsr_get_data_p_z, dbcsr_scalar_are_equal, dbcsr_scalar_negative, &
dbcsr_scalar_one, dbcsr_type_1d_to_2d
USE dbcsr_data_types, ONLY: dbcsr_datatype_sizeof
USE dbcsr_dist_methods, ONLY: dbcsr_distribution_col_dist,&
dbcsr_distribution_has_threads,&
@ -2540,8 +2540,15 @@ CONTAINS
product_matrix_size_comm => product_matrix_size_1
product_matrix_size_calc => product_matrix_size_2
! Reuse right buffers for the recv product matrices
! Use original data
product_matrix_recv_comm%matrix => buffers_comm%right%buffer%mats(1)
CALL dbcsr_data_release(product_matrix_recv_comm%matrix%m%data_area)
product_matrix_recv_comm%matrix%m%data_area = buffers_orig%right%data(1)
CALL dbcsr_data_hold(product_matrix_recv_comm%matrix%m%data_area)
product_matrix_recv_calc%matrix => buffers_calc%right%buffer%mats(1)
CALL dbcsr_data_release(product_matrix_recv_calc%matrix%m%data_area)
product_matrix_recv_calc%matrix%m%data_area = buffers_orig%left%data(1)
CALL dbcsr_data_hold(product_matrix_recv_calc%matrix%m%data_area)
CALL dbcsr_data_init(data_send)
CALL dbcsr_data_new(data_send, data_type)
DO iproc = 1, nranks3D+1