mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-04 13:41:24 +00:00
ffe9775e70
-Added parameter to enable/disable verification and timing of kernel in various examples that missed it. -Added parameter to change number of groups to execute in grouped_gemm_examples. Signed-off-by: Michal Kulikowski <Michal.Kulikowski@amd.com>
40 lines
1.7 KiB
C++
40 lines
1.7 KiB
C++
// SPDX-License-Identifier: MIT
|
|
// Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
#include "common.hpp"
|
|
|
|
using InDataType = F16;
|
|
using OutDataType = F16;
|
|
|
|
// clang-format off
|
|
using DevicePermuteInstance = ck::tensor_operation::device::DevicePermuteImpl
|
|
// ######| NumDim| InData| OutData| Elementwise| Block| NPer| HPer| WPer| InBlock| InBlockTransfer| InBlockTransfer| Src| Dst| Src| Dst|
|
|
// ######| | Type| Type| Operation| Size| Block| Block| Block| LdsExtraW| ThreadClusterLengths| ThreadClusterArrangeOrder| VectorDim| VectorDim| ScalarPerVector| ScalarPerVector|
|
|
// ######| | | | | | | | | | | | | | | |
|
|
// ######| | | | | | | | | | | | | | | |
|
|
< 3, InDataType, OutDataType, PassThrough, 256, 1, 32, 32, 3, S<1, 32, 8>, S<0, 1, 2>, 2, 1, 2, 1>;
|
|
// clang-format on
|
|
|
|
#include "run_permute_element_example.inc"
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
bool time_kernel = false;
|
|
|
|
if(argc == 1)
|
|
{
|
|
// use default
|
|
}
|
|
else if(argc == 2)
|
|
{
|
|
time_kernel = std::stoi(argv[1]);
|
|
}
|
|
else
|
|
{
|
|
printf("arg1: time kernel (0=no, 1=yes, default=0)\n");
|
|
exit(0);
|
|
}
|
|
|
|
return !run_permute_element_example({1, 32000, 80}, {0, 2, 1}, time_kernel);
|
|
}
|