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
153 lines
4.5 KiB
YAML
153 lines
4.5 KiB
YAML
GlobalParameters:
|
|
BoundsCheck: False
|
|
KernelTime: True
|
|
PrintSolutionRejectionReason: True
|
|
|
|
BenchmarkProblems:
|
|
|
|
########################################
|
|
# NN / TLUA = 1, TLUB = 0
|
|
########################################
|
|
-
|
|
- # ProblemType
|
|
OperationType: GEMM
|
|
DataType: s
|
|
TransposeA: False
|
|
TransposeB: False
|
|
UseBeta: True
|
|
Batched: True
|
|
|
|
- # BenchmarkProblemSizeGroup
|
|
InitialSolutionParameters:
|
|
BenchmarkCommonParameters:
|
|
- BufferLoad: [True]
|
|
- BufferStore: [True]
|
|
- KernelLanguage: ["Assembly"]
|
|
- EdgeType: ["ShiftPtr"]
|
|
- LoopTail: [True]
|
|
- WorkGroupMapping: [8]
|
|
ForkParameters:
|
|
- VectorWidth: [2]
|
|
- ThreadTile:
|
|
- [ 2, 8 ]
|
|
- WorkGroup:
|
|
- [ 16, 8, 1 ] # 32x64
|
|
- DepthU: [32,64]
|
|
BenchmarkForkParameters:
|
|
JoinParameters:
|
|
BenchmarkJoinParameters:
|
|
BenchmarkFinalParameters:
|
|
- ProblemSizes:
|
|
- Range: [ [16777216], [32], [1], [32] ] # strideA = 16777216, should fail on DU >= 64 (For tensorA-N)
|
|
- Range: [ [32], [32], [1], [16777216] ] # strideB = 16777216, should fail on MT1 >= 64 (For tensorB-N)
|
|
|
|
########################################
|
|
# NT / TLUA = 1, TLUB = 1
|
|
########################################
|
|
-
|
|
- # ProblemType
|
|
OperationType: GEMM
|
|
DataType: s
|
|
TransposeA: False
|
|
TransposeB: True
|
|
UseBeta: True
|
|
Batched: True
|
|
|
|
- # BenchmarkProblemSizeGroup
|
|
InitialSolutionParameters:
|
|
BenchmarkCommonParameters:
|
|
- BufferLoad: [True]
|
|
- BufferStore: [True]
|
|
- KernelLanguage: ["Assembly"]
|
|
- EdgeType: ["ShiftPtr"]
|
|
- LoopTail: [True]
|
|
- WorkGroupMapping: [8]
|
|
ForkParameters:
|
|
- VectorWidth: [2]
|
|
- ThreadTile:
|
|
- [ 2, 2 ]
|
|
- WorkGroup:
|
|
- [ 16, 16, 1 ] # 32x32
|
|
- DepthU: [32,64]
|
|
BenchmarkForkParameters:
|
|
JoinParameters:
|
|
BenchmarkJoinParameters:
|
|
BenchmarkFinalParameters:
|
|
- ProblemSizes:
|
|
- Range: [ [16777216], [32], [1], [32] ] # strideA = 16777216, should fail on DU >= 64 (For tensorA-N)
|
|
- Range: [ [32], [16777216], [1], [32] ] # strideB = 16777216, should fail on DU >= 64 (For tensorB-T)
|
|
|
|
########################################
|
|
# TN / TLUA = 0, TLUB = 0
|
|
########################################
|
|
-
|
|
- # ProblemType
|
|
OperationType: GEMM
|
|
DataType: s
|
|
TransposeA: True
|
|
TransposeB: False
|
|
UseBeta: True
|
|
Batched: True
|
|
|
|
- # BenchmarkProblemSizeGroup
|
|
InitialSolutionParameters:
|
|
BenchmarkCommonParameters:
|
|
- BufferLoad: [True]
|
|
- BufferStore: [True]
|
|
- KernelLanguage: ["Assembly"]
|
|
- EdgeType: ["ShiftPtr"]
|
|
- LoopTail: [True]
|
|
- WorkGroupMapping: [8]
|
|
ForkParameters:
|
|
- VectorWidth: [2]
|
|
- ThreadTile:
|
|
- [ 2, 2 ]
|
|
- WorkGroup:
|
|
- [ 16, 16, 1 ] # 32x32
|
|
- [ 16, 32, 1 ] # 32x64
|
|
- [ 32, 16, 1 ] # 64x32
|
|
- DepthU: [16] # DU doesn't affect bufferload limit in TN
|
|
BenchmarkForkParameters:
|
|
JoinParameters:
|
|
BenchmarkJoinParameters:
|
|
BenchmarkFinalParameters:
|
|
- ProblemSizes:
|
|
- Range: [ [32], [32], [1], [16777216] ] # strideA = 16777216, should fail on MT0 >= 64 (For tensorA-T) # strideB = 16777216, should fail on MT1 >= 64 (For tensorB-N)
|
|
|
|
########################################
|
|
# TT / TLUA = 0, TLUB = 1
|
|
########################################
|
|
-
|
|
- # ProblemType
|
|
OperationType: GEMM
|
|
DataType: s
|
|
TransposeA: True
|
|
TransposeB: True
|
|
UseBeta: True
|
|
Batched: True
|
|
|
|
- # BenchmarkProblemSizeGroup
|
|
InitialSolutionParameters:
|
|
BenchmarkCommonParameters:
|
|
- BufferLoad: [True]
|
|
- BufferStore: [True]
|
|
- KernelLanguage: ["Assembly"]
|
|
- EdgeType: ["ShiftPtr"]
|
|
- LoopTail: [True]
|
|
- WorkGroupMapping: [8]
|
|
ForkParameters:
|
|
- VectorWidth: [2]
|
|
- ThreadTile:
|
|
- [ 2, 2 ]
|
|
- WorkGroup:
|
|
- [ 16, 8, 1 ] # 32x16
|
|
- [ 32, 8, 1 ] # 64x16
|
|
- DepthU: [32,64]
|
|
BenchmarkForkParameters:
|
|
JoinParameters:
|
|
BenchmarkJoinParameters:
|
|
BenchmarkFinalParameters:
|
|
- ProblemSizes:
|
|
- Range: [ [32], [32], [1], [16777216] ] # strideA = 16777216, should fail on MT0 >= 64 (For tensorA-T)
|
|
- Range: [ [32], [16777216], [1], [32] ] # strideB = 16777216, should fail on DU >= 64 (For tensorB-T)
|