---
include: hipblas_common.yaml

Definitions:
  - &N_range
    - [ -1, 0, 1000 ]

  - &incx_incy_range
    - { incx:  1, incy:  1 }
    - { incx: -1, incy: -2 }

  - &alpha_beta_range
    - { alpha: 2.0, alphai:  2.0, beta:  0.0, betai: 0.0 }

  - &batch_count_range
    - [ -1, 0, 5 ]

Tests:
  - name: axpy_general
    category: quick
    function: axpy
    precision: *single_double_precisions_complex_real
    alpha_beta: *alpha_beta_range
    N: *N_range
    incx_incy: *incx_incy_range
    api: [ FORTRAN, C, FORTRAN_64, C_64 ]

  # half precision only rocBLAS backend
  - name: axpy_general_half
    category: quick
    function: axpy
    precision: *half_precision
    alpha_beta: *alpha_beta_range
    N: *N_range
    incx_incy: *incx_incy_range
    api: [ FORTRAN, C, FORTRAN_64, C_64 ]
    backend_flags: AMD

  - name: axpy_batched_general
    category: quick
    function: axpy_batched
    precision: *single_double_precisions_complex_real_half_real
    alpha_beta: *alpha_beta_range
    N: *N_range
    incx_incy: *incx_incy_range
    batch_count: *batch_count_range
    api: [ FORTRAN, C, FORTRAN_64, C_64 ]
    backend_flags: AMD

  - name: axpy_strided_batched_general
    category: quick
    function: axpy_strided_batched
    precision: *single_double_precisions_complex_real_half_real
    alpha_beta: *alpha_beta_range
    N: *N_range
    incx_incy: *incx_incy_range
    batch_count: *batch_count_range
    stride_scale: [ 2.5 ]
    api: [ FORTRAN, C, FORTRAN_64, C_64 ]
    backend_flags: AMD

  # ILP-64 tests
  # - name: axpy_64
  #   category: stress
  #   function:
  #     - axpy: *single_precision
  #   arguments:
  #     - { N: 2147483649, incx:  1, incy:  1 }
  #     - { N: 2, incx:  2147483649, incy:  1 }
  #     - { N: 2, incx:  1, incy:  2147483649 }
  #   api: [ C_64 ]
  #   os_flags: [ LINUX ]
  #   gpu_arch: '90a'

  # - name: axpy_64
  #   category: stress
  #   function:
  #     - axpy_batched: *single_precision
  #     - axpy_strided_batched: *single_precision
  #   arguments:
  #     - { N: 2147483649, incx:  1, incy:  1, batch_count: 1 }
  #     - { N: 2, incx:  2147483649, incy:  1, batch_count: 1 }
  #     - { N: 2, incx:  1, incy:  2147483649, batch_count: 1 }
  #     - { N: 2, incx:  1, incy:  1, stride_x: 2, stride_y: 2, batch_count: 666666 }
  #   api: [ C_64 ]
  #   os_flags: [ LINUX ]
  #   gpu_arch: '90a'

  - name: axpy_bad_arg
    category: pre_checkin
    function:
      - axpy_bad_arg
    precision: *single_double_precisions_complex_real
    api: [ FORTRAN, C, FORTRAN_64, C_64 ]
    backend_flags: AMD

  - name: axpy_bad_arg
    category: pre_checkin
    function:
      - axpy_bad_arg
    precision: *single_double_precisions_complex_real
    api: [ FORTRAN, C, FORTRAN_64, C_64 ]
    bad_arg_all: false
    backend_flags: NVIDIA

  - name: axpy_bad_arg
    category: pre_checkin
    function:
        - axpy_bad_arg: *half_precision
        - axpy_batched_bad_arg: *single_double_precisions_complex_real_half_real
        - axpy_strided_batched_bad_arg: *single_double_precisions_complex_real_half_real
    api: [ FORTRAN, C, FORTRAN_64, C_64 ]
    backend_flags: AMD
...