------------------------------------------------------------------------------ -- -- -- GNAT RUN-TIME COMPONENTS -- -- -- -- S Y S T E M . G E N E R I C _ R E A L _ B L A S -- -- -- -- B o d y -- -- -- -- Copyright (C) 2006-2009, Free Software Foundation, Inc. -- -- -- -- GNAT is free software; you can redistribute it and/or modify it under -- -- terms of the GNU General Public License as published by the Free Soft- -- -- ware Foundation; either version 3, or (at your option) any later ver- -- -- sion. GNAT is distributed in the hope that it will be useful, but WITH- -- -- OUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY -- -- or FITNESS FOR A PARTICULAR PURPOSE. -- -- -- -- As a special exception under Section 7 of GPL version 3, you are granted -- -- additional permissions described in the GCC Runtime Library Exception, -- -- version 3.1, as published by the Free Software Foundation. -- -- -- -- You should have received a copy of the GNU General Public License and -- -- a copy of the GCC Runtime Library Exception along with this program; -- -- see the files COPYING3 and COPYING.RUNTIME respectively. If not, see -- -- . -- -- -- -- GNAT was originally developed by the GNAT team at New York University. -- -- Extensive contributions were provided by Ada Core Technologies Inc. -- -- -- ------------------------------------------------------------------------------ with Ada.Unchecked_Conversion; use Ada; with Interfaces; use Interfaces; with Interfaces.Fortran; use Interfaces.Fortran; with Interfaces.Fortran.BLAS; use Interfaces.Fortran.BLAS; with System.Generic_Array_Operations; use System.Generic_Array_Operations; package body System.Generic_Real_BLAS is Is_Single : constant Boolean := Real'Machine_Mantissa = Fortran.Real'Machine_Mantissa and then Fortran.Real (Real'First) = Fortran.Real'First and then Fortran.Real (Real'Last) = Fortran.Real'Last; Is_Double : constant Boolean := Real'Machine_Mantissa = Double_Precision'Machine_Mantissa and then Double_Precision (Real'First) = Double_Precision'First and then Double_Precision (Real'Last) = Double_Precision'Last; -- Local subprograms function To_Double_Precision (X : Real) return Double_Precision; pragma Inline_Always (To_Double_Precision); function To_Real (X : Double_Precision) return Real; pragma Inline_Always (To_Real); -- Instantiations function To_Double_Precision is new Vector_Elementwise_Operation (X_Scalar => Real, Result_Scalar => Double_Precision, X_Vector => Real_Vector, Result_Vector => Double_Precision_Vector, Operation => To_Double_Precision); function To_Real is new Vector_Elementwise_Operation (X_Scalar => Double_Precision, Result_Scalar => Real, X_Vector => Double_Precision_Vector, Result_Vector => Real_Vector, Operation => To_Real); function To_Double_Precision is new Matrix_Elementwise_Operation (X_Scalar => Real, Result_Scalar => Double_Precision, X_Matrix => Real_Matrix, Result_Matrix => Double_Precision_Matrix, Operation => To_Double_Precision); function To_Real is new Matrix_Elementwise_Operation (X_Scalar => Double_Precision, Result_Scalar => Real, X_Matrix => Double_Precision_Matrix, Result_Matrix => Real_Matrix, Operation => To_Real); function To_Double_Precision (X : Real) return Double_Precision is begin return Double_Precision (X); end To_Double_Precision; function To_Real (X : Double_Precision) return Real is begin return Real (X); end To_Real; --------- -- dot -- --------- function dot (N : Positive; X : Real_Vector; Inc_X : Integer := 1; Y : Real_Vector; Inc_Y : Integer := 1) return Real is begin if Is_Single then declare type X_Ptr is access all BLAS.Real_Vector (X'Range); type Y_Ptr is access all BLAS.Real_Vector (Y'Range); function Conv_X is new Unchecked_Conversion (Address, X_Ptr); function Conv_Y is new Unchecked_Conversion (Address, Y_Ptr); begin return Real (sdot (N, Conv_X (X'Address).all, Inc_X, Conv_Y (Y'Address).all, Inc_Y)); end; elsif Is_Double then declare type X_Ptr is access all BLAS.Double_Precision_Vector (X'Range); type Y_Ptr is access all BLAS.Double_Precision_Vector (Y'Range); function Conv_X is new Unchecked_Conversion (Address, X_Ptr); function Conv_Y is new Unchecked_Conversion (Address, Y_Ptr); begin return Real (ddot (N, Conv_X (X'Address).all, Inc_X, Conv_Y (Y'Address).all, Inc_Y)); end; else return Real (ddot (N, To_Double_Precision (X), Inc_X, To_Double_Precision (Y), Inc_Y)); end if; end dot; ---------- -- gemm -- ---------- procedure gemm (Trans_A : access constant Character; Trans_B : access constant Character; M : Positive; N : Positive; K : Positive; Alpha : Real := 1.0; A : Real_Matrix; Ld_A : Integer; B : Real_Matrix; Ld_B : Integer; Beta : Real := 0.0; C : in out Real_Matrix; Ld_C : Integer) is begin if Is_Single then declare subtype A_Type is BLAS.Real_Matrix (A'Range (1), A'Range (2)); subtype B_Type is BLAS.Real_Matrix (B'Range (1), B'Range (2)); type C_Ptr is access all BLAS.Real_Matrix (C'Range (1), C'Range (2)); function Conv_A is new Unchecked_Conversion (Real_Matrix, A_Type); function Conv_B is new Unchecked_Conversion (Real_Matrix, B_Type); function Conv_C is new Unchecked_Conversion (Address, C_Ptr); begin sgemm (Trans_A, Trans_B, M, N, K, Fortran.Real (Alpha), Conv_A (A), Ld_A, Conv_B (B), Ld_B, Fortran.Real (Beta), Conv_C (C'Address).all, Ld_C); end; elsif Is_Double then declare subtype A_Type is Double_Precision_Matrix (A'Range (1), A'Range (2)); subtype B_Type is Double_Precision_Matrix (B'Range (1), B'Range (2)); type C_Ptr is access all Double_Precision_Matrix (C'Range (1), C'Range (2)); function Conv_A is new Unchecked_Conversion (Real_Matrix, A_Type); function Conv_B is new Unchecked_Conversion (Real_Matrix, B_Type); function Conv_C is new Unchecked_Conversion (Address, C_Ptr); begin dgemm (Trans_A, Trans_B, M, N, K, Double_Precision (Alpha), Conv_A (A), Ld_A, Conv_B (B), Ld_B, Double_Precision (Beta), Conv_C (C'Address).all, Ld_C); end; else declare DP_C : Double_Precision_Matrix (C'Range (1), C'Range (2)); begin if Beta /= 0.0 then DP_C := To_Double_Precision (C); end if; dgemm (Trans_A, Trans_B, M, N, K, Double_Precision (Alpha), To_Double_Precision (A), Ld_A, To_Double_Precision (B), Ld_B, Double_Precision (Beta), DP_C, Ld_C); C := To_Real (DP_C); end; end if; end gemm; ---------- -- gemv -- ---------- procedure gemv (Trans : access constant Character; M : Natural := 0; N : Natural := 0; Alpha : Real := 1.0; A : Real_Matrix; Ld_A : Positive; X : Real_Vector; Inc_X : Integer := 1; Beta : Real := 0.0; Y : in out Real_Vector; Inc_Y : Integer := 1) is begin if Is_Single then declare subtype A_Type is BLAS.Real_Matrix (A'Range (1), A'Range (2)); subtype X_Type is BLAS.Real_Vector (X'Range); type Y_Ptr is access all BLAS.Real_Vector (Y'Range); function Conv_A is new Unchecked_Conversion (Real_Matrix, A_Type); function Conv_X is new Unchecked_Conversion (Real_Vector, X_Type); function Conv_Y is new Unchecked_Conversion (Address, Y_Ptr); begin sgemv (Trans, M, N, Fortran.Real (Alpha), Conv_A (A), Ld_A, Conv_X (X), Inc_X, Fortran.Real (Beta), Conv_Y (Y'Address).all, Inc_Y); end; elsif Is_Double then declare subtype A_Type is Double_Precision_Matrix (A'Range (1), A'Range (2)); subtype X_Type is Double_Precision_Vector (X'Range); type Y_Ptr is access all Double_Precision_Vector (Y'Range); function Conv_A is new Unchecked_Conversion (Real_Matrix, A_Type); function Conv_X is new Unchecked_Conversion (Real_Vector, X_Type); function Conv_Y is new Unchecked_Conversion (Address, Y_Ptr); begin dgemv (Trans, M, N, Double_Precision (Alpha), Conv_A (A), Ld_A, Conv_X (X), Inc_X, Double_Precision (Beta), Conv_Y (Y'Address).all, Inc_Y); end; else declare DP_Y : Double_Precision_Vector (Y'Range); begin if Beta /= 0.0 then DP_Y := To_Double_Precision (Y); end if; dgemv (Trans, M, N, Double_Precision (Alpha), To_Double_Precision (A), Ld_A, To_Double_Precision (X), Inc_X, Double_Precision (Beta), DP_Y, Inc_Y); Y := To_Real (DP_Y); end; end if; end gemv; ---------- -- nrm2 -- ---------- function nrm2 (N : Natural; X : Real_Vector; Inc_X : Integer := 1) return Real is begin if Is_Single then declare subtype X_Type is BLAS.Real_Vector (X'Range); function Conv_X is new Unchecked_Conversion (Real_Vector, X_Type); begin return Real (snrm2 (N, Conv_X (X), Inc_X)); end; elsif Is_Double then declare subtype X_Type is Double_Precision_Vector (X'Range); function Conv_X is new Unchecked_Conversion (Real_Vector, X_Type); begin return Real (dnrm2 (N, Conv_X (X), Inc_X)); end; else return Real (dnrm2 (N, To_Double_Precision (X), Inc_X)); end if; end nrm2; end System.Generic_Real_BLAS;