---
include: hipsparselt_common.yaml
include: known_bugs.yaml
include: spmm_strided_batched_common.yaml

Definitions:
  - &transA_transB_range
    - { transA: N, transB: N }
    - { transA: T, transB: T }

  - &alpha_beta_range
    - { alpha:  1.0, beta:  0.0 }

Tests:
- name: compress_strided_batched_small
  category: quick
  function:
    compress_strided_batched: *real_precisions_2b
  matrix_size: *strided_batched_small_matrix_size_range
  alpha_beta: *alpha_beta_range
  transA_transB: *transA_transB_range
  batch_count: [ 3 ]
  sparse_b: [ true, false]

- name: compress_strided_batched_small_stride_zero
  category: quick
  function:
    compress_strided_batched: *real_precisions_2b
  matrix_size: *strided_batched_small_matrix_size_stride_a_range
  transA: N
  transB: N
  batch_count: [ 1, 3 ]
  sparse_b: [ true, false]

- name: compress_strided_batched_medium
  category: pre_checkin
  function:
    compress_strided_batched: *real_precisions_2b
  matrix_size: *strided_batched_medium_matrix_size_range
  transA_transB: *transA_transB_range
  alpha_beta: *alpha_beta_range
  batch_count: [ 3 ]
  sparse_b: [ true, false]
  func_version: [1, 2]

- name: compress_strided_batched_medium_alt
  category: pre_checkin
  function:
    compress_strided_batched: *hpa_half_precision
  matrix_size: *strided_batched_medium_matrix_size_range
  transA_transB: *transA_transB_range
  alpha: 1
  beta: 0
  initialization: special
  batch_count: [ 3 ]
  sparse_b: [ true, false]

- name: compress_strided_batched_medium_stride_zero
  category: nightly
  function:
    compress_strided_batched: *real_precisions_2b
  matrix_size: *strided_batched_medium_matrix_size_stride_a_range
  transA: N
  transB: N
  batch_count: 3
  sparse_b: [ true, false]

- name: compress_strided_batched_medium_stride_zero_alt
  category: nightly
  function:
    compress_strided_batched: *hpa_half_precision
  matrix_size: *strided_batched_medium_matrix_size_stride_a_range
  alpha: 1
  beta: 0
  initialization: special
  transA: N
  transB: N
  batch_count: 3
  sparse_b: [ true, false]

- name: compress_strided_batched_large
  category: pre_checkin
  function:
    compress_strided_batched: *real_precisions_2b
  matrix_size: *strided_batched_large_matrix_size_range
  transA_transB: *transA_transB_range
  alpha_beta: *alpha_beta_range
  batch_count: [ 3 ]
  sparse_b: [ true, false]

- name: compress_strided_batched_large_stride_zero
  category: pre_checkin
  function:
    compress_strided_batched: *real_precisions_2b
  matrix_size: *strided_batched_large_matrix_size_stride_a_range
  transA: N
  transB: N
  batch_count: [ 3 ]
  sparse_b: [ true, false]

- name: compress_strided_batched_deepbench
  category: nightly
  function:
    compress_strided_batched: *real_precisions_2b
  matrix_size: *strided_batched_deepbench_vec
  alpha: 1.0
  beta: 0.0
  transA: N
  transB: T
  batch_count: [ 2 ]
  sparse_b: [ true, false]

- name: compress_strided_batched_deepbench_alt
  category: nightly
  function:
    compress_strided_batched: *hpa_half_precision
  matrix_size: *strided_batched_deepbench_vec
  alpha: 1.0
  beta: 0.0
  initialization: special
  transA: N
  transB: T
  batch_count: [ 2 ]
  sparse_b: [ true, false]

- name: compress_strided_batched_conv_resnet50_bwddata
  category: nightly
  function:
    compress_strided_batched: *real_precisions_2b
  matrix_size: *strided_batched_conv_resnet50_bwddata
  alpha: 1.0
  beta: 0.0
  transA: N
  transB: T
  batch_count: 4
  sparse_b: [ true, false]

- name: compress_strided_batched_conv_inception4_bwddata
  category: nightly
  function:
    compress_strided_batched: *real_precisions_2b
  matrix_size: *strided_batched_conv_inception4_bwddata
  alpha: 1.0
  beta: 0.0
  transA: N
  transB: T
  batch_count: 4
  sparse_b: [ true, false]

- name: compress_strided_batched_conv_ctest_fwd
  category: nightly
  function:
    compress_strided_batched: *real_precisions_2b
  matrix_size: *strided_batched_conv_ctest_fwd
  alpha: 1.0
  beta: 0.0
  transA: N
  transB: N
  batch_count: 1
  sparse_b: [ true, false]

- name: compress_strided_batched_smoke
  category: smoke
  transA_transB: *transA_transB_range
  function:
    compress_strided_batched: *real_precisions_2b
  matrix_size:
    - { M: 8, N: 8, K: 16 }
    - { M: 8, N: 8, K: 32 }
    - { M: 8, N: 8, K: 48 }
    - { M: 8, N: 8, K: 64 }
    - { M: 8, N: 8, K: 128 }
    - { M: 8, N: 8, K: 192 }
    - { M: 8, N: 8, K: 256 }
    - { M: 128, N: 128, K: 16 }
    - { M: 128, N: 128, K: 24 }
    - { M: 128, N: 128, K: 32 }
    - { M: 128, N: 128, K: 48 }
    - { M: 128, N: 128, K: 64 }
    - { M: 128, N: 128, K: 128 }
    - { M: 128, N: 128, K: 192 }
    - { M: 128, N: 128, K: 256 }
  sparse_b: [ true, false]
  batch_count: [1, 3]
  func_version: [1, 2]
...