Mercurial > octave
view liboctave/external/blas-xtra/dmatm3.f @ 27919:1891570abac8
update Octave Project Developers copyright for the new year
In files that have the "Octave Project Developers" copyright notice,
update for 2020.
author | John W. Eaton <jwe@octave.org> |
---|---|
date | Mon, 06 Jan 2020 22:29:51 -0500 |
parents | b442ec6dda5c |
children | bd51beb6205e |
line wrap: on
line source
c Copyright (C) 2009-2020 The Octave Project Developers c c See the file COPYRIGHT.md in the top-level directory of this distribution c or <https://octave.org/COPYRIGHT.html/>. c c c Author: Jaroslav Hajek <highegg@gmail.com> c c This file is part of Octave. c c Octave is free software: you can redistribute it and/or modify it c under the terms of the GNU General Public License as published by c the Free Software Foundation, either version 3 of the License, or c (at your option) any later version. c c Octave is distributed in the hope that it will be useful, but c WITHOUT ANY WARRANTY; without even the implied warranty of c MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the c GNU General Public License for more details. c c You should have received a copy of the GNU General Public License c along with Octave; see the file COPYING. If not, see c <https://www.gnu.org/licenses/>. c subroutine dmatm3(m,n,k,np,a,b,c) c purpose: a 3-dimensional matrix product. c given a (m,k,np) array a and (k,n,np) array b, c calculates a (m,n,np) array c such that c for i = 1:np c c(:,:,i) = a(:,:,i) * b(:,:,i) c c arguments: c m,n,k (in) the dimensions c np (in) number of multiplications c a (in) a double prec. input array, size (m,k,np) c b (in) a double prec. input array, size (k,n,np) c c (out) a double prec. output array, size (m,n,np) integer m,n,k,np double precision a(m*k,np),b(k*n,np) double precision c(m*n,np) double precision ddot,one,zero parameter (one = 1d0, zero = 0d0) external ddot,dgemv,dgemm integer i c quick return if possible. if (np <= 0) return if (m == 1) then if (n == 1) then do i = 1,np c(1,i) = ddot(k,a(1,i),1,b(1,i),1) end do else do i = 1,np call dgemv("T",k,n,one,b(1,i),k,a(1,i),1,zero,c(1,i),1) end do end if else if (n == 1) then do i = 1,np call dgemv("N",m,k,one,a(1,i),m,b(1,i),1,zero,c(1,i),1) end do else do i = 1,np call dgemm("N","N",m,n,k, + one,a(1,i),m,b(1,i),k,zero,c(1,i),m) end do end if end if end subroutine