Skip to content

Commit

Permalink
Fix garbage print
Browse files Browse the repository at this point in the history
  • Loading branch information
mkrack committed May 31, 2022
1 parent 1b05dea commit 96161d2
Show file tree
Hide file tree
Showing 2 changed files with 23 additions and 23 deletions.
13 changes: 7 additions & 6 deletions src/dbt/tas/dbt_tas_mm.F
Original file line number Diff line number Diff line change
Expand Up @@ -216,13 +216,14 @@ RECURSIVE SUBROUTINE dbt_tas_multiply(transa, transb, transc, alpha, matrix_a, m
dims_b = [dbt_tas_nblkrows_total(matrix_b), dbt_tas_nblkcols_total(matrix_b)]
dims_c = [dbt_tas_nblkrows_total(matrix_c), dbt_tas_nblkcols_total(matrix_c)]

IF (unit_nr_prv .GT. 0) THEN
WRITE (unit_nr_prv, '(A)') REPEAT("-", 80)
WRITE (unit_nr_prv, '(A,1X,A,1X,A,1X,A,1X,A,1X,A)') "DBT TAS MATRIX MULTIPLICATION:", &
TRIM(dbm_get_name(matrix_a%matrix)), 'x', &
TRIM(dbm_get_name(matrix_b%matrix)), '=', &
IF (unit_nr_prv > 0) THEN
WRITE (unit_nr_prv, "(A)") REPEAT("-", 80)
WRITE (unit_nr_prv, "(A)") &
"DBT TAS MATRIX MULTIPLICATION: "// &
TRIM(dbm_get_name(matrix_a%matrix))//" x "// &
TRIM(dbm_get_name(matrix_b%matrix))//" = "// &
TRIM(dbm_get_name(matrix_c%matrix))
WRITE (unit_nr_prv, '(A)') REPEAT("-", 80)
WRITE (unit_nr_prv, "(A)") REPEAT("-", 80)
END IF
IF (do_batched) THEN
IF (unit_nr_prv > 0) THEN
Expand Down
33 changes: 16 additions & 17 deletions src/dbt/tas/dbt_tas_test.F
Original file line number Diff line number Diff line change
Expand Up @@ -125,7 +125,7 @@ SUBROUTINE dbt_tas_setup_test_matrix(matrix, mp_comm_out, mp_comm, nrows, ncols,
cbsize_obj = dbt_tas_blk_size_arb(cbsizes)
CALL dbt_tas_distribution_new(dist, mp_comm_out, row_dist_obj, col_dist_obj)
CALL dbt_tas_create(matrix, name, dist=dist, &
CALL dbt_tas_create(matrix, name=TRIM(name), dist=dist, &
row_blk_size=rbsize_obj, col_blk_size=cbsize_obj, own_dist=.TRUE.)
max_row_size = MAXVAL(rbsizes)
Expand Down Expand Up @@ -324,13 +324,14 @@ SUBROUTINE dbt_tas_test_mm(transa, transb, transc, matrix_a, matrix_b, matrix_c,
REAL(KIND=dp), PARAMETER :: test_tol = 1.0E-10_dp
CHARACTER(LEN=8) :: status_str
INTEGER :: comm_dbm, io_unit, mp_comm, mynode, &
numnodes
INTEGER, CONTIGUOUS, DIMENSION(:), POINTER :: cd_a, cd_b, cd_c, col_block_sizes_a, &
col_block_sizes_b, col_block_sizes_c, rd_a, rd_b, rd_c, row_block_sizes_a, &
row_block_sizes_b, row_block_sizes_c
INTEGER, DIMENSION(2) :: myploc, npdims
LOGICAL :: transa_prv, transb_prv
LOGICAL :: abort, transa_prv, transb_prv
REAL(KIND=dp) :: norm, rc_cs, sq_cs
TYPE(dbm_distribution_obj) :: dist_a, dist_b, dist_c
TYPE(dbm_type) :: dbm_a, dbm_a_mm, dbm_b, dbm_b_mm, dbm_c, &
Expand All @@ -342,6 +343,7 @@ SUBROUTINE dbt_tas_test_mm(transa, transb, transc, matrix_a, matrix_b, matrix_c,
CALL dbt_tas_get_split_info(dbt_tas_info(matrix_a), mp_comm=mp_comm)
CALL mp_environ(numnodes, mynode, mp_comm)
abort = .FALSE.
io_unit = -1
IF (mynode .EQ. 0) io_unit = unit_nr
Expand Down Expand Up @@ -431,22 +433,19 @@ SUBROUTINE dbt_tas_test_mm(transa, transb, transc, matrix_a, matrix_b, matrix_c,
norm = dbm_maxabs(dbm_c_mm_check)
IF (io_unit > 0) THEN
IF (ABS(norm) .GT. test_tol) THEN
WRITE (io_unit, '(A, A, A, A, A, 1X, A)') TRIM(dbm_get_name(matrix_a%matrix)), &
transa, ' X ', &
TRIM(dbm_get_name(matrix_b%matrix)), &
transb, 'failed!'
WRITE (io_unit, "(A,1X,E9.2,1X,E9.2)") "checksums", sq_cs, rc_cs
WRITE (io_unit, "(A,1X,E9.2)") "difference norm", norm
CPABORT("")
ELSE
WRITE (io_unit, '(A, A, A, A, A, 1X, A)') TRIM(dbm_get_name(matrix_a%matrix)), &
transa, ' X ', &
TRIM(dbm_get_name(matrix_b%matrix)), &
transb, 'passed!'
IF (ABS(norm) > test_tol) THEN
status_str = " failed!"
abort = .TRUE.
ELSE
status_str = " passed!"
abort = .FALSE.
END IF
WRITE (io_unit, "(A)") &
TRIM(dbm_get_name(matrix_a%matrix))//" x "// &
TRIM(dbm_get_name(matrix_b%matrix))//TRIM(status_str)
WRITE (io_unit, "(A,1X,E9.2,1X,E9.2)") "checksums", sq_cs, rc_cs
WRITE (io_unit, "(A,1X,E9.2)") "difference norm", norm
END IF
IF (abort) CPABORT("DBT TAS test failed")
END IF
CALL dbm_release(dbm_a)
Expand All @@ -463,7 +462,7 @@ SUBROUTINE dbt_tas_test_mm(transa, transb, transc, matrix_a, matrix_b, matrix_c,
CALL mp_comm_free(comm_dbm)
END SUBROUTINE
END SUBROUTINE dbt_tas_test_mm
! **************************************************************************************************
!> \brief Calculate checksum of tall-and-skinny matrix consistent with dbm_checksum
Expand Down

0 comments on commit 96161d2

Please sign in to comment.