mirror of
https://github.com/sfilippone/psblas3.git
synced 2026-10-07 15:14:57 +00:00
[UPDATE] Updated psb_spmm_test file, adding functionalities to read input data from file and write output data on fle. Also added a first README.md file, stil to be finished for documentation
This commit is contained in:
Stack-1
+23
-16
@@ -5,31 +5,38 @@ include $(INCDIR)/Make.inc.psblas
|
||||
#
|
||||
# Libraries used
|
||||
#
|
||||
LIBDIR=$(INSTALLDIR)/lib/
|
||||
PSBLAS_LIB= -L$(LIBDIR) -lpsb_util -lpsb_prec -lpsb_base
|
||||
LDLIBS=$(PSBLDLIBS)
|
||||
LIBDIR = $(INSTALLDIR)/lib/
|
||||
PSBLAS_LIB = -L$(LIBDIR) -lpsb_util -lpsb_prec -lpsb_base
|
||||
LDLIBS = $(PSBLDLIBS)
|
||||
|
||||
FINCLUDES=$(FMFLAG)$(MODDIR) $(FMFLAG).
|
||||
|
||||
OBJS=psb_spmm_test.o
|
||||
|
||||
EXEDIR=./runs
|
||||
|
||||
all: runsd psb_spmm_test
|
||||
|
||||
runsd:
|
||||
(if test ! -d runs ; then mkdir runs; fi)
|
||||
GREEN=\033[0;32m
|
||||
RED=\033[0;31m
|
||||
BLUE=\033[0;34m
|
||||
YELLOW=\033[33m
|
||||
END_COLOUR=\033[0m
|
||||
|
||||
psb_spmm_test: $(OBJS)
|
||||
$(FLINK) $(LOPT) $(OBJS) -o psb_spmm_test $(PSBLAS_LIB) $(LDLIBS)
|
||||
/bin/mv psb_spmm_test $(EXEDIR)
|
||||
|
||||
all: runsd psb_spmm_test
|
||||
@printf "$(GREEN)[INFO]\tCompilation success!$(END_COLOUR)\n"
|
||||
|
||||
runsd:
|
||||
@(if test ! -d runs ; then mkdir runs; fi)
|
||||
@printf "$(BLUE)[INFO]\tBuild directory $(EXEDIR) correctly initialized$(END_COLOUR)\n"
|
||||
|
||||
|
||||
psb_spmm_test:
|
||||
@$(FLINK) $(LOPT) psb_spmm_test.f90 -o psb_spmm_test -I../../modules/ -I. $(PSBLAS_LIB) $(LDLIBS)
|
||||
@mv psb_spmm_test $(EXEDIR)
|
||||
@printf "$(BLUE)[INFO]\tTesting files generated correctly$(END_COLOUR)\n"
|
||||
|
||||
clean:
|
||||
/bin/rm -f $(OBJS)\
|
||||
@rm -f $(OBJS)\
|
||||
*$(.mod) $(EXEDIR)/psb_spmm_test
|
||||
|
||||
lib:
|
||||
(cd ../../; make library)
|
||||
verycleanlib:
|
||||
(cd ../../; make veryclean)
|
||||
.PHONY: all runsd clean
|
||||
|
||||
|
||||
+39
-2
@@ -1,8 +1,45 @@
|
||||
# Introduction
|
||||
This is a directory developed by Luca Pepè Sciarria and Simone Staccone froma Tor Vergata University to start to create some unit tests for PSBLAS 3.9, in particular for psb_spmm routine.
|
||||
|
||||
## GEtting started
|
||||
## Environment
|
||||
These tests are developed using a linux environment, in particular Rocky Linux 9.5 (Blue Onyx).
|
||||
|
||||
The compiler used is:
|
||||
- gnu 12.2.1
|
||||
|
||||
The necessary dependnces are:
|
||||
- mpich 4.2.2
|
||||
- psblas 3.9
|
||||
|
||||
In order to have the exact same environment used for testing compile PSBALS library using cuda 12.5.
|
||||
|
||||
|
||||
## Getting started
|
||||
Steps to reproduce the tests:
|
||||
-
|
||||
- make
|
||||
- insert the matrix files inside the matrix/ directory (or create one if it doesn't exists; psblas3/test/spmm/matrix/)
|
||||
- run ./runs/psb_spmm_test
|
||||
- ...
|
||||
|
||||
## Test goal
|
||||
Check the correctness of the matrix-vector multiplication $y = Ax$ using the **psb_spmm** routine, checking for all the test suite cases.
|
||||
|
||||
## Test Suite
|
||||
**A** matrixes are located in the matrix/ directory
|
||||
|Matrix|File Name|Type|
|
||||
|:-:|:-:|:-:|
|
||||
|$A_1$|1138_bus.mtx|Sparse|
|
||||
|
||||
**x** vectors are located in the vectors/ directory. They are generated randomly using the same seed and then saved on different files based on their characteristics. The size of the vector is choosen accordingly to the size of the matrix column space considered for the single test instance.
|
||||
|Vector|File Name|Coefficients|Coefficients Description|
|
||||
|:-:|:-:|:-:|:-:|
|
||||
|$x_1$|x1.txt|$x_i> 0, \forall i$|Positive coefficients|
|
||||
|$x_2$|x2.txt|$x_i < 0, \forall i$|Negative coefficients
|
||||
|$x_3$|x3.txt|$x_i \ne 0, \forall i$|Random coefficients
|
||||
|$x_4$|x4.txt|$x_i = 0, \forall i$|Null coefficients
|
||||
|
||||
## Output
|
||||
The results of the computation will be saved on different files based on the instance of the test considered. In particular the naming conventiona format the output file as sol_m#_x#_y#.mtx, where each # is a number choosen w.r.t. the test instance. (Ex. sol_m1_x1_y1.mtx is the solution computed using the first matrix file, the first x vector file and the first y vector file).
|
||||
|
||||
## Notes
|
||||
For now only integer multiplication is tested and on a single matrix
|
||||
+122
-43
@@ -1,40 +1,45 @@
|
||||
! Test program for y = AX spmm routine
|
||||
! Author: Luca Pepé Sciarria, Staccone Simone (Tor Vergata University)
|
||||
!> Test program for y = AX spsb_pmm routine
|
||||
!! Check the README.md to see all details about the tests.
|
||||
!!
|
||||
!! Author: Luca Pepé Sciarria, Staccone Simone (Tor Vergata University)
|
||||
program psb_spmm_test
|
||||
use psb_base_mod
|
||||
use psb_prec_mod
|
||||
use psb_util_mod
|
||||
|
||||
implicit none
|
||||
|
||||
! input parameters
|
||||
character(len=40) :: mtx_file, file_name, name
|
||||
character(len=256) :: mtx_file, file_name, name
|
||||
|
||||
! Testing parameters
|
||||
character(len=256),dimension(2) :: mtx_files
|
||||
|
||||
! sparse matrices
|
||||
type(psb_sspmat_type) :: a
|
||||
type(psb_lsspmat_type) :: aux_a
|
||||
type(psb_sspmat_type) :: a
|
||||
type(psb_lsspmat_type) :: aux_a
|
||||
|
||||
! vectors
|
||||
type(psb_s_vect_type) :: x, y
|
||||
type(psb_s_vect_type) :: x, y
|
||||
|
||||
! matrix descriptor data structure
|
||||
type(psb_desc_type) :: desc_a
|
||||
type(psb_desc_type) :: desc_a
|
||||
|
||||
! communication context
|
||||
type(psb_ctxt_type) :: ctxt
|
||||
integer(psb_ipk_) :: my_rank, np, info, err_act
|
||||
type(psb_ctxt_type) :: ctxt
|
||||
integer(psb_ipk_) :: my_rank, np, info, err_act
|
||||
|
||||
! matrix parameters
|
||||
integer(psb_ipk_) :: m, n, nnz
|
||||
integer(psb_ipk_) :: nr, nt ! In BLOCK ROWS distributin, the number of rows
|
||||
integer(psb_ipk_) :: m, n, nnz
|
||||
integer(psb_ipk_) :: nr, nt ! In BLOCK ROWS distributin, the number of rows
|
||||
|
||||
|
||||
real(psb_spk_), allocatable :: x_global(:)
|
||||
integer(psb_ipk_) :: i
|
||||
real(psb_spk_), allocatable :: x_global(:), y_global(:)
|
||||
integer(psb_ipk_) :: i
|
||||
|
||||
name = "psb_spmm_test" ! Name of the program to output in case of error
|
||||
info = psb_success_
|
||||
mtx_file = "matrix/1138_bus.mtx"
|
||||
mtx_files(1) = "matrix/1138_bus.mtx"
|
||||
|
||||
|
||||
call psb_init(ctxt)
|
||||
@@ -57,7 +62,10 @@ program psb_spmm_test
|
||||
goto 9999
|
||||
end if
|
||||
|
||||
call psb_matdist(aux_a, a, ctxt,desc_a,info,fmt="COO",parts=part_block)
|
||||
|
||||
! part_block it's a macro defined in psb_blockpart_mod to identify BLOCK ROWS distribution
|
||||
call psb_matdist(aux_a, a, ctxt,desc_a,info,fmt="COO",parts=part_block)
|
||||
|
||||
|
||||
m = aux_a%get_nrows()
|
||||
n = aux_a%get_ncols()
|
||||
@@ -76,31 +84,13 @@ program psb_spmm_test
|
||||
|
||||
! Generate random array for b using always the same seed
|
||||
if(my_rank == psb_root_) then
|
||||
call random_init(repeatable=.true.,image_distinct=.false.)
|
||||
allocate(x_global(n))
|
||||
call random_number(x_global)
|
||||
allocate(y_global(n))
|
||||
call generate_vectors(n) ! True for x
|
||||
call mm_array_read(x_global,info,filename="vectors/x1.mtx")
|
||||
call mm_array_read(y_global,info,filename="vectors/y1.mtx")
|
||||
end if
|
||||
|
||||
! nt = (m+np-1)/np
|
||||
! nr = max(0,min(nt,m-(my_rank*nt)))
|
||||
|
||||
! Check if distribution metadata is correct
|
||||
! nt = nr
|
||||
! call psb_sum(ctxt,nt)
|
||||
! if (nt /= m) then
|
||||
! write(psb_err_unit,*) my_rank, 'Initialization error ',nr,nt,m
|
||||
! info = -1
|
||||
! call psb_barrier(ctxt)
|
||||
! call psb_abort(ctxt)
|
||||
! return
|
||||
! end if
|
||||
|
||||
! call psb_cdall(ctxt,desc_a,info, nl=nr)
|
||||
! if(info /= psb_success_) then
|
||||
! write(psb_out_unit,*) "Error in dexcriptor allocator routine using BLOCK ROWS distribution"
|
||||
! goto 9999
|
||||
! end if
|
||||
|
||||
call psb_geall(x,desc_a,info)
|
||||
if(info /= psb_success_) then
|
||||
write(psb_out_unit,*) "Error allocating x data structure"
|
||||
@@ -113,19 +103,32 @@ program psb_spmm_test
|
||||
write(psb_out_unit,*) "Error in psb_scatter to populate x data structure"
|
||||
goto 9999
|
||||
end if
|
||||
|
||||
|
||||
call psb_geall(y,desc_a,info)
|
||||
if(info /= psb_success_) then
|
||||
write(psb_out_unit,*) "Error allocating y data structure"
|
||||
goto 9999
|
||||
end if
|
||||
|
||||
call y%zero()
|
||||
|
||||
! Populate y class using data from y_global vector
|
||||
call psb_scatter(y_global,y,desc_a,info,root=psb_root_)
|
||||
if(info /= psb_success_) then
|
||||
write(psb_out_unit,*) "Error in psb_scatter to populate y data structure"
|
||||
goto 9999
|
||||
end if
|
||||
|
||||
|
||||
! y = alpha * A * x + betha * y
|
||||
call psb_spmm(sone,a,x,sone,y,desc_a,info)
|
||||
if(info /= psb_success_) then
|
||||
write(psb_out_unit,*) "Error in psb_spmm routine"
|
||||
goto 9999
|
||||
end if
|
||||
|
||||
! Save result to output file
|
||||
call mm_array_write(y,"Result vector",info,filename="sol_m1_x1_y1.mtx")
|
||||
|
||||
! Deallocate
|
||||
call psb_gefree(x, desc_a,info)
|
||||
if(info /= psb_success_) then
|
||||
@@ -151,9 +154,8 @@ program psb_spmm_test
|
||||
goto 9999
|
||||
end if
|
||||
|
||||
if(my_rank == psb_root_) then
|
||||
deallocate(x_global)
|
||||
end if
|
||||
deallocate(x_global)
|
||||
deallocate(y_global)
|
||||
|
||||
call psb_exit(ctxt)
|
||||
stop
|
||||
@@ -167,4 +169,81 @@ program psb_spmm_test
|
||||
call psb_exit(ctxt)
|
||||
stop
|
||||
|
||||
end program
|
||||
|
||||
contains
|
||||
|
||||
!> @brief Function to randomly generate x and y vectors
|
||||
!! and save them on multiple files based on their
|
||||
!! coefficients values.
|
||||
!!
|
||||
!! @param n The size of the vector.
|
||||
subroutine generate_vectors(n)
|
||||
use psb_base_mod
|
||||
|
||||
implicit none
|
||||
|
||||
integer(psb_ipk_), intent(in) :: n
|
||||
integer(psb_ipk_), allocatable :: x(:), y(:)
|
||||
real(psb_spk_), allocatable :: v(:)
|
||||
integer(psb_ipk_) :: i
|
||||
|
||||
allocate(x(n))
|
||||
allocate(y(n))
|
||||
allocate(v(2*n))
|
||||
|
||||
|
||||
call random_init(repeatable=.true.,image_distinct=.true.)
|
||||
call random_number(v)
|
||||
|
||||
do i = 1,n
|
||||
x(i) = int(v(i) * 100)
|
||||
end do
|
||||
|
||||
do i = 1,n
|
||||
y(i) = int(v(i+n) * 100)
|
||||
end do
|
||||
|
||||
|
||||
! Write only positive in x_1
|
||||
call mm_array_write(x,"Positive vector",info,filename="vectors/x1.mtx")
|
||||
call mm_array_write(y,"Positive vector",info,filename="vectors/y1.mtx")
|
||||
|
||||
! Write only negative in x_2
|
||||
do i=1,n
|
||||
x(i) = -x(i)
|
||||
y(i) = -y(i)
|
||||
end do
|
||||
|
||||
call mm_array_write(x,"Negative vector",info,filename="vectors/x2.mtx")
|
||||
call mm_array_write(y,"Negative vector",info,filename="vectors/y2.mtx")
|
||||
|
||||
|
||||
! Since numbers are less than one and always positive, we have to generate negative ones subtractiong 50
|
||||
do i=1,n
|
||||
x(i) = -x(i) ! Make the values positive again
|
||||
x(i) = x(i) - 50
|
||||
y(i) = -y(i) ! Make the values positive again
|
||||
y(i) = y(i) - 50
|
||||
end do
|
||||
|
||||
! Write random in x_3
|
||||
call mm_array_write(x,"Random vector",info,filename="vectors/x3.mtx")
|
||||
call mm_array_write(y,"Random vector",info,filename="vectors/y3.mtx")
|
||||
|
||||
! Write zero in x_4
|
||||
do i=1,n
|
||||
x(i) = 0
|
||||
y(i) = 0
|
||||
end do
|
||||
|
||||
call mm_array_write(x,"Null vector",info,filename="vectors/x4.mtx")
|
||||
call mm_array_write(y,"Null vector",info,filename="vectors/y4.mtx")
|
||||
|
||||
deallocate(x)
|
||||
deallocate(y)
|
||||
deallocate(v)
|
||||
|
||||
end subroutine
|
||||
|
||||
end program
|
||||
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
Reference in New Issue
Block a user