derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
213 lines
5.1 KiB
YAML
213 lines
5.1 KiB
YAML
---
|
|
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: prune_strided_batched_small
|
|
category: quick
|
|
function:
|
|
prune_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 ]
|
|
prune_algo: [ 0, 1 ]
|
|
sparse_b: [true, false]
|
|
|
|
- name: prune_strided_batched_small_stride_zero
|
|
category: quick
|
|
function:
|
|
prune_strided_batched: *real_precisions_2b
|
|
matrix_size: *strided_batched_small_matrix_size_stride_a_range
|
|
transA: N
|
|
transB: N
|
|
batch_count: [ 1, 3 ]
|
|
prune_algo: [ 0, 1 ]
|
|
sparse_b: [true, false]
|
|
|
|
- name: prune_strided_batched_medium
|
|
category: pre_checkin
|
|
function:
|
|
prune_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 ]
|
|
prune_algo: [ 0, 1 ]
|
|
sparse_b: [true, false]
|
|
func_version: [1, 2]
|
|
|
|
- name: prune_strided_batched_medium_alt
|
|
category: pre_checkin
|
|
function:
|
|
prune_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 ]
|
|
prune_algo: [ 0, 1 ]
|
|
sparse_b: [true, false]
|
|
|
|
- name: prune_strided_batched_medium_stride_zero
|
|
category: nightly
|
|
function:
|
|
prune_strided_batched: *real_precisions_2b
|
|
matrix_size: *strided_batched_medium_matrix_size_stride_a_range
|
|
transA: N
|
|
transB: N
|
|
batch_count: 3
|
|
prune_algo: [ 0, 1 ]
|
|
sparse_b: [true, false]
|
|
|
|
- name: prune_strided_batched_medium_stride_zero_alt
|
|
category: nightly
|
|
function:
|
|
prune_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
|
|
prune_algo: [ 0, 1 ]
|
|
sparse_b: [true, false]
|
|
|
|
- name: prune_strided_batched_large
|
|
category: pre_checkin
|
|
function:
|
|
prune_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 ]
|
|
prune_algo: [ 0, 1 ]
|
|
sparse_b: [true, false]
|
|
|
|
- name: prune_strided_batched_large_stride_zero
|
|
category: pre_checkin
|
|
function:
|
|
prune_strided_batched: *real_precisions_2b
|
|
matrix_size: *strided_batched_large_matrix_size_stride_a_range
|
|
transA: N
|
|
transB: N
|
|
batch_count: [ 3 ]
|
|
prune_algo: [ 0, 1 ]
|
|
sparse_b: [true, false]
|
|
|
|
- name: prune_strided_batched_deepbench
|
|
category: nightly
|
|
function:
|
|
prune_strided_batched: *real_precisions_2b
|
|
matrix_size: *strided_batched_deepbench_vec
|
|
alpha: 1.0
|
|
beta: 0.0
|
|
transA: N
|
|
transB: T
|
|
batch_count: [ 2 ]
|
|
prune_algo: [ 0, 1 ]
|
|
sparse_b: [true, false]
|
|
|
|
- name: prune_strided_batched_deepbench_alt
|
|
category: nightly
|
|
function:
|
|
prune_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 ]
|
|
prune_algo: [ 0, 1 ]
|
|
sparse_b: [true, false]
|
|
|
|
- name: prune_strided_batched_conv_resnet50_fwd
|
|
category: nightly
|
|
function:
|
|
prune_strided_batched: *real_precisions_2b
|
|
matrix_size: *strided_batched_conv_resnet50_fwd
|
|
alpha: 1.0
|
|
beta: 0.0
|
|
transA: N
|
|
transB: N
|
|
batch_count: 4
|
|
prune_algo: [ 0, 1 ]
|
|
sparse_b: [true, false]
|
|
|
|
- name: prune_strided_batched_conv_resnet50_bwddata
|
|
category: nightly
|
|
function:
|
|
prune_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
|
|
prune_algo: [ 0, 1 ]
|
|
sparse_b: [true, false]
|
|
|
|
- name: prune_strided_batched_conv_inception4_bwddata
|
|
category: nightly
|
|
function:
|
|
prune_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
|
|
prune_algo: [ 0, 1 ]
|
|
sparse_b: [true, false]
|
|
|
|
- name: prune_strided_batched_conv_ctest_fwd
|
|
category: nightly
|
|
function:
|
|
prune_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
|
|
prune_algo: [ 0, 1 ]
|
|
sparse_b: [true, false]
|
|
|
|
- name: prune_strided_batched_smoke
|
|
category: smoke
|
|
transA_transB: *transA_transB_range
|
|
function:
|
|
prune_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]
|
|
prune_algo: [ 0, 1 ]
|
|
...
|