Skip to content

Commit 34b5c81

Browse files
committed
[hipblaslt] implement tests for 64-bit offset support
1 parent 3749ee4 commit 34b5c81

5 files changed

Lines changed: 583 additions & 2 deletions

File tree

projects/hipblaslt/clients/common/include/hipblaslt_arguments.hpp

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,12 @@ struct Arguments
112112
int32_t batch_count;
113113
int32_t batch_mode;
114114

115+
// Batch offset support for general batched GEMM
116+
int64_t batch_offset_a;
117+
int64_t batch_offset_b;
118+
int64_t batch_offset_c;
119+
int64_t batch_offset_d;
120+
115121
int32_t iters;
116122
int32_t cold_iters;
117123

@@ -236,6 +242,10 @@ struct Arguments
236242
OPER(lde) SEP \
237243
OPER(batch_count) SEP \
238244
OPER(batch_mode) SEP \
245+
OPER(batch_offset_a) SEP \
246+
OPER(batch_offset_b) SEP \
247+
OPER(batch_offset_c) SEP \
248+
OPER(batch_offset_d) SEP \
239249
OPER(iters) SEP \
240250
OPER(cold_iters) SEP \
241251
OPER(algo) SEP \

0 commit comments

Comments
 (0)