From df497fc915b6a50ea13e0b755e86ac7f1b54fc01 Mon Sep 17 00:00:00 2001 From: Wojtek Kosior Date: Tue, 23 Apr 2019 22:19:35 +0200 Subject: implement naive multiplication --- src/naivemath.F90 | 60 ++++++++++++++++++++++++++++++++++++++++++++++++++++++- 1 file changed, 59 insertions(+), 1 deletion(-) (limited to 'src') diff --git a/src/naivemath.F90 b/src/naivemath.F90 index ef33aea..a83505e 100644 --- a/src/naivemath.F90 +++ b/src/naivemath.F90 @@ -38,6 +38,64 @@ MODULE naivmat CONTAINS - + FUNCTION naivmull_4(A, B) result(C) + IMPLICIT none + real(kind=4), intent(in), dimension(1:,1:) :: A, B + real(kind=4), dimension(:,:), allocatable :: C + integer :: i, j, k + + allocate(C(size(A, 1), size(B, 2))) + C = 0 + + DO i = 1, size(A(:,1)) + DO j = 1, size(B(1,:)) + DO k = 1, size(A(1,:)) + + C(i,j) = C(i,j) + A(i,k) * B(k,j) + END DO + END DO + END DO + + END FUNCTION naivmull_4 + + FUNCTION naivmull_8(A, B) result(C) + IMPLICIT none + real(kind=8), intent(in), dimension(1:,1:) :: A, B + real(kind=8), dimension(:,:), allocatable :: C + integer :: i, j, k + + allocate(C(size(A, 1), size(B, 2))) + C = 0 + + DO i = 1, size(A, 1) + DO j = 1, size(B, 2) + DO k = 1, size(A, 2) + + C(i,j) = C(i,j) + A(i,k) * B(k,j) + END DO + END DO + END DO + + END FUNCTION naivmull_8 + + FUNCTION naivmull_16(A, B) result(C) + IMPLICIT none + real(kind=16), intent(in), dimension(1:,1:) :: A, B + real(kind=16), dimension(:,:), allocatable :: C + integer :: i, j, k + + allocate(C(size(A, 1), size(B, 2))) + C = 0 + + DO i = 1, size(A, 1) + DO j = 1, size(B, 2) + DO k = 1, size(A, 2) + + C(i,j) = C(i,j) + A(i,k) * B(k,j) + END DO + END DO + END DO + + END FUNCTION naivmull_16 END MODULE naivmat -- cgit v1.2.3