From 51ce1efb441b6950032d622e6f859797db256fa7 Mon Sep 17 00:00:00 2001 From: Wojtek Kosior Date: Tue, 23 Apr 2019 22:25:27 +0200 Subject: implement better multiplication --- src/bettermath.F90 | 59 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 59 insertions(+) diff --git a/src/bettermath.F90 b/src/bettermath.F90 index 4b64c54..e2c0773 100644 --- a/src/bettermath.F90 +++ b/src/bettermath.F90 @@ -38,5 +38,64 @@ MODULE bettmat CONTAINS + FUNCTION bettmull_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 j = 1, size(B(1,:)) + DO k = 1, size(A(1,:)) + DO i = 1, size(A(:,1)) + + C(i,j) = C(i,j) + A(i,k) * B(k,j) + END DO + END DO + END DO + + END FUNCTION bettmull_4 + + FUNCTION bettmull_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 j = 1, size(B(1,:)) + DO k = 1, size(A(1,:)) + DO i = 1, size(A(:,1)) + + C(i,j) = C(i,j) + A(i,k) * B(k,j) + END DO + END DO + END DO + + END FUNCTION bettmull_8 + FUNCTION bettmull_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 j = 1, size(B(1,:)) + DO k = 1, size(A(1,:)) + DO i = 1, size(A(:,1)) + + C(i,j) = C(i,j) + A(i,k) * B(k,j) + END DO + END DO + END DO + + END FUNCTION bettmull_16 + END MODULE bettmat -- cgit v1.2.3