[rocm-libraries] ROCm/rocm-libraries#5348 (commit 7b18234)

[CK][Examples] Adding parameters for a couple of CK examples:
 -gemm_add_add_mean_meansquare_xdl_fp16 -gemm_dl_quantization_int8
 -gemm_xdl_bias_relu_quantization_int8 -gemm_xdl_quantization_int8

Signed-off-by: Michal Kulikowski <Michal.Kulikowski@amd.com>
This commit is contained in:
Michał Kulikowski
2026-03-12 08:48:36 +00:00
committed by assistant-librarian[bot]
parent a1679e38ee
commit 2c3f9bfa52
4 changed files with 121 additions and 7 deletions

View File

@@ -99,7 +99,7 @@ using DeviceGemmInstance = ck::tensor_operation::device::DeviceGemmMultipleD_Xdl
using ReferenceGemmInstance = ck::tensor_operation::host::
ReferenceGemm<ADataType, BDataType, EDataType, float, PassThrough, PassThrough, CDEElementOp>;
int main()
int main(int argc, char* argv[])
{
bool do_verification = true;
bool time_kernel = false;
@@ -115,6 +115,34 @@ int main()
float requant_scale = 0.03;
if(argc == 1)
{
// do nothing
}
else if(argc == 3 || argc == 9)
{
do_verification = std::stoi(argv[1]);
time_kernel = std::stoi(argv[2]);
if(argc == 9)
{
M = std::stoi(argv[3]);
N = std::stoi(argv[4]);
K = std::stoi(argv[5]);
StrideA = std::stoi(argv[6]);
StrideB = std::stoi(argv[7]);
StrideE = std::stoi(argv[8]);
}
}
else
{
std::cout << "arg1: verification (0=no, 1=yes)\n"
<< " arg2: Measure kernel execution time (1=ON, 0=Off)\n"
<< " arg3 to 8: M (256x), N(128x), K(64x), StrideA, StrideB, StrideE\n"
<< std::endl;
exit(1);
}
auto f_host_tensor_descriptor =
[](std::size_t row, std::size_t col, std::size_t stride, auto layout) {
using namespace ck::literals;