mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-12 01:10:17 +00:00
7965d66a81
* spolit the static library into several * update lib paths and fix client example * do not use device_mha_operarions for client examples * use appropriate libs to link to client examples * remove the gpu/transpose path from the list * try fixing clinet examples 3,4,9 * add necessary libs for client examples * fix the layernorm client example * fix the client examples 23 and 24 * fix typo * add interface library and refresh clang format
29 lines
647 B
C++
29 lines
647 B
C++
#pragma once
|
|
|
|
TYPED_TEST(TestTranspose, Test1)
|
|
{
|
|
// for 16, 8, 16, 32, 8
|
|
std::vector<int> Ms{1, 2, 3, 4, 5, 6};
|
|
std::vector<index_t> lengths{16, 8, 16, 32, 8};
|
|
/**constexpr int N = 16;
|
|
constexpr int C = 8;
|
|
constexpr int D = 16;
|
|
constexpr int H = 32;
|
|
constexpr int W = 8;**/
|
|
|
|
this->Run();
|
|
}
|
|
|
|
TYPED_TEST(TestTranpose, Test2)
|
|
{
|
|
std::vector<int> Ms{127, 255, 312, 799, 1573};
|
|
std::vector<index_t> lengths{16, 8, 16, 32, 16};
|
|
/**constexpr int N = 16;
|
|
constexpr int C = 8;
|
|
constexpr int D = 16;
|
|
constexpr int H = 32;
|
|
constexpr int W = 8;**/
|
|
|
|
this->Run();
|
|
}
|