add test case in extend for large K correctness
@@ -115,6 +115,22 @@ BenchmarkProblems:
- ProblemSizes:
- Range: [ [16], [16], [1], [16777216] ] # strideA = 16777216, should fail on MT0 >= 32 (For tensorA-T) # strideB = 16777216, should fail on MT1 >= 32 (For tensorB-N)
+ - BenchmarkCommonParameters:
+ BenchmarkFinalParameters:
+ ForkParameters:
+ - EdgeType: [ShiftPtr]
+ - KernelLanguage: [Assembly]
+ - ThreadTile:
+ - [4, 4]
+ - WorkGroup:
+ - [16, 16, 1]
+ - DepthU: [4]
+ - GlobalSplitU: [1, 10]
+ BenchmarkJoinParameters:
+ BenchmarkFinalParameters:
+ - ProblemSizes:
+ - Exact: [250, 250, 1, 8000000]
+
########################################
# TT / TLUA = 0, TLUB = 1
########################################
