!** Copyright (c) 1989, NVIDIA CORPORATION.  All rights reserved.
!**
!** Licensed under the Apache License, Version 2.0 (the "License");
!** you may not use this file except in compliance with the License.
!** You may obtain a copy of the License at
!**
!**     http://www.apache.org/licenses/LICENSE-2.0
!**
!** Unless required by applicable law or agreed to in writing, software
!** distributed under the License is distributed on an "AS IS" BASIS,
!** WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
!** See the License for the specific language governing permissions and
!** limitations under the License.

!* Tests for runtime library MATMUL routines

program p

  parameter(NbrTests=208)

  real*4, dimension(4,3) :: arr1
  real*4, dimension(3) :: arr2
  real*4, dimension(4) :: arr3
  real*4, dimension(4,4) :: arr4
  real*4, dimension(0:3,-1:1) :: arr5
  real*4, dimension(-3:-1) :: arr6
  real*4, dimension(-1:2,0:3) :: arr7
  real*4, dimension(2:5,3) :: arr8
  real*4, dimension(2:4) :: arr9
  real*4, dimension(2:5) :: arr10
  real*4, dimension(4,2:4) :: arr11
  real*4, dimension(1:4) :: arr12

  data arr1 /0,1,2,3,4,5,6,7,8,9,10,11/
  data arr5 /0,1,2,3,4,5,6,7,8,9,10,11/
  data arr2 /0,1,2/
  data arr6 /0,1,2/
  data arr3 /0,1,2,3/
  data arr4 /0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15/
  data arr7 /0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15/
  data arr8 /0,1,2,3,4,5,6,7,8,9,10,11/
  data arr9 /0,1,2/
  data arr10 /0,1,2,3/
  data arr11 /0,1,2,3,4,5,6,7,8,9,10,11/

  real*4 :: expect(NbrTests)
  real*4 :: results(NbrTests)

  data expect /  &
  ! tests 1-4
      20.0, 23.0, 26.0, 29.0,  &
  ! tests 5-8
      0.0, 23.0, 26.0, 29.0,  &
  ! tests 9-12
      23.0, 26.0, 29.0, 0.0,  &
  ! tests 13-16
      0.0, 20.0, 23.0, 26.0,  &
  ! tests 17-20
      0.0, 5.0, 6.0, 7.0,  &
  ! tests 21-24
      8.0, 9.0, 10.0, 11.0,  &
  ! tests 25-28
      8.0, 11.0, 14.0, 17.0,  &
  ! tests 29-32
      0.0, 8.0, 11.0, 14.0,  &
  ! tests 33-36
      9.0, 10.0, 11.0, 0.0,  &
  ! tests 37-40
      18.0, 20.0, 22.0, 0.0,  &
  ! tests 41-44
      0.0, 8.0, 0.0, 14.0,  &
  ! tests 45-48
      9.0, 0.0, 11.0,  &
      0.0,  &
  ! tests 49-64
      0.0, 9.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 11.0, 0.0, 0.0,  &
      0.0, 0.0, 0.0, 0.0,  &
  ! tests 65-80
      0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 9.0, 0.0, 11.0, 0.0,  &
      0.0, 0.0, 0.0, 0.0,  &
  ! tests 81-96
      0.0, 9.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 11.0, 0.0, 0.0,  &
      0.0, 0.0, 0.0, 0.0,  &
  ! tests 97-112
      0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 23.0, 0.0, 29.0, 0.0,  &
      0.0, 0.0, 0.0, 0.0,  &
  ! tests 113-116
      22.0, 20.0, 18.0, 0.0,  &
  ! tests 117-120
      9.0, 0.0, 11.0, 0.0,  &
  ! tests 121-136
      0.0, 11.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 9.0, 0.0, 0.0,  &
      0.0, 0.0, 0.0, 0.0,  &
  ! tests 137-152
      0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 9.0, 0.0, 11.0, 0.0,  &
      0.0, 0.0, 0.0, 0.0,  &
  ! tests 153-168
      0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 23.0, 0.0, 29.0, 0.0,  &
      0.0, 0.0, 0.0, 0.0,  &
  ! tests 169-172
      18.0, 20.0, 22.0, 0.0, &
  ! tests 173-176
     20.0, 23.0, 26.0, 29.0,  &
  ! tests  177-180
     20.0, 23.0, 26.0, 29.0,  &
  ! tests 181-184
      23.0, 26.0, 29.0, 0.0,  &
  ! tests 185-188
      0.0, 20.0, 23.0, 26.0,  &
  ! tests 189-192
      0.0, 5.0, 6.0, 7.0,  &
  ! tests 193-196
      8.0, 9.0, 10.0, 11.0,  &
  ! tests 197-200
      8.0, 11.0, 14.0, 17.0,  &
  ! tests 201-204
      0.0, 8.0, 11.0, 14.0,  &
  ! tests 205-208
      9.0, 10.0, 11.0, 0.0/

  results = -1

  ! tests 1-4
  arr3=0
  arr3 = matmul(arr1,arr2)
  call assign_result(1,4,arr3,results)
  !print *,arr3
  
  ! tests 5-8
  arr3=0
  arr3(2:4) = matmul(arr1(2:4,:),arr2)
  call assign_result(5,8,arr3,results)
  !print *,arr3
  
  ! tests 9-12
  arr3=0
  arr3(1:3) = matmul(arr1(2:4,:),arr2)
  call assign_result(9,12,arr3,results)
  !print *,arr3
  
  !tests 13-16
  arr3=0
  arr3(2:4) = matmul(arr1(1:3,:),arr2)
  call assign_result(13,16,arr3,results)
  !print *,arr3
  
  !tests 17-20
  arr3=0
  arr3(2:4) = matmul(arr1(2:4,1:2),arr2(1:2))
  call assign_result(17,20,arr3,results)
  !print *,arr3
  
  !tests 21-24
  arr3=0
  arr3 = matmul(arr1(:,2:3),arr2(1:2))
  call assign_result(21,24,arr3,results)
  !print *,arr3
  
  !tests 25-28
  arr3=0
  arr3 = matmul(arr1(:,1:2),arr2(2:3))
  call assign_result(25,28,arr3,results)
  !print *,arr3
  
  !tests 29-32
  arr3=0
  arr3(2:4)  = matmul(arr1(1:3,1:2),arr2(2:3))
  call assign_result(29,32,arr3,results)
  !print *,arr3
  
  !tests 33-36
  arr3=0
  arr3(1:3)  = matmul(arr1(2:4,2:3),arr2(1:2))
  call assign_result(33,36,arr3,results)
  !print *,arr3
  
  !tests 37-40
  arr3=0
  arr3(1:3) = matmul(arr1(2:4,1:3:2),arr2(1:3:2))
  call assign_result(37,40,arr3,results)
  !print *,arr3
  
  !tests 41-44
  arr3=0
  arr3(2:4:2)  = matmul(arr1(1:3:2,1:2),arr2(2:3))
  call assign_result(41,44,arr3,results)
  !print *,arr3
  
  !tests 45-48
  arr3=0
  arr3(1:3:2)  = matmul(arr1(2:4:2,2:3),arr2(1:2))
  call assign_result(45,48,arr3,results)
  !print *,arr3
  
  !tests 49-64
  arr4=0
  arr4(2,1:3:2)  = matmul(arr1(2:4:2,2:3),arr2(1:2))
  call assign_result(49,64,arr4,results)
  !print *,arr4
  
  !tests 65-80
  arr4=0
  arr4(1:3:2,3)  = matmul(arr1(2:4:2,2:3),arr2(1:2))
  call assign_result(65,80,arr4,results)
  !print *,arr4
  
  !tests 81-96
  arr7=0
  arr7(0,0:2:2)  = matmul(arr5(1:3:2,0:1),arr6(-3:-2))
  call assign_result(81,96,arr7,results)
  !print *,arr7
  
  !tests 97-112
  arr7=0
  arr7(-1:1:2,2)  = matmul(arr5(1:3:2,0:1),arr6(-2:-1))
  call assign_result(97,112,arr7,results)
  !print *,arr7
  
  !tests 113-116
  arr3=0
  arr3(3:1:-1) = matmul(arr1(2:4,3:1:-2),arr2(3:1:-2))
  call assign_result(113,116,arr3,results)
  !print *,arr3
  
  !tests 117-120
  arr3=0
  arr3(3:1:-2)  = matmul(arr1(4:2:-2,2:3),arr2(1:2))
  call assign_result(117,120,arr3,results)
  !print *,arr3
  
  !tests 121,136
  arr4=0
  arr4(2,3:1:-2)  = matmul(arr1(2:4:2,2:3),arr2(1:2))
  call assign_result(121,136,arr4,results)
  !print *,arr4
  
  !tests 137-152
  arr4=0
  arr4(3:1:-2,3)  = matmul(arr1(4:2:-2,2:3),arr2(1:2))
  call assign_result(137,152,arr4,results)
  !print *,arr4
  
  !tests 153-168
  arr7=0
  arr7(1:-1:-2,2)  = matmul(arr5(3:1:-2,0:1),arr6(-2:-1))
  call assign_result(153,168,arr7,results)
  !print *,arr7
  
  !tests 169-172
  arr3=0
  arr3(1:3) = matmul(arr1(2:4,3:1:-2),arr2(3:1:-2))
  call assign_result(169,172,arr3,results)
  !print *,arr3

  arr12 = 0

  !print *,"tests 173-176"
  arr10=0
  arr10 = arr12 + matmul(arr8,arr9) 
  call assign_result(173,176,arr10,results)
  !print *,arr10

  !print *,"tests 177-180"
  arr10=0
  arr10 = arr12 +  matmul(arr11,arr9) 
  call assign_result(177,180,arr10,results)
  !print *,arr10

  !print *,"tests 181-184"
  arr10=0
  arr10(2:4) = arr12(2:4) + matmul(arr8(3:5,:),arr9)
  call assign_result(181,184,arr10,results)
  !print *,arr10
  
  !print *,"tests 185-188"
  arr10=0 
  arr10(3:5) = arr12(2:4) + matmul(arr8(2:4,:),arr9)
  call assign_result(185,188,arr10,results)
  !print *,arr10
  
  !print *,"tests 189-192"
  arr10=0 
  arr10(3:5) = arr12(2:4) + matmul(arr8(3:5,1:2),arr9(2:3))
  call assign_result(189,192,arr10,results)
  !print *,arr10
  
  !print *,"tests 193-196"
  arr10=0 
  arr10 = arr12 + matmul(arr8(:,2:3),arr9(2:3))
  call assign_result(193,196,arr10,results)
  !print *,arr10

  !print *,"tests 197-200"
  arr10=0
  arr10 = arr12 + matmul(arr8(:,1:2),arr9(3:4))
  call assign_result(197,200,arr10,results)
  !print *,arr10

  !print *,"tests 201-204"
  arr10=0
  arr10(3:5)  = arr12(2:4) + matmul(arr8(2:4,1:2),arr9(3:4))
  call assign_result(201,204,arr10,results)
  !print *,arr10

  !print *,"tests 205-208"
  arr10=0
  arr10(2:4)  = arr12(2:4) + matmul(arr8(3:5,2:3),arr9(2:3))
  call assign_result(205,208,arr10,results)
  !print *,arr10

  call check(results, expect, NbrTests)

end program

subroutine assign_result(s_idx, e_idx , arr, rslt)
  REAL*4, dimension(1:e_idx-s_idx+1) :: arr
  REAL*4, dimension(e_idx) :: rslt
  integer:: s_idx, e_idx

  rslt(s_idx:e_idx) = arr

end subroutine