Skip to content

GitLab

"tests/test_config.py" did not exist on "5311f564ed1e1247f83d83533059006845071463"
Commit 0c5f9438 authored by Adam Osewski's avatar Adam Osewski
Browse files

Constepxr everything!

parent 01f0831b
Hide whitespace changes
Inline Side-by-side
Showing with 115 additions and 115 deletions
test/gemm_split_k/test_gemm_splitk_ut_cases.inc
View file @ 0c5f9438
......@@ -3,12 +3,12 @@
TYPED_TEST(TestGemmSplitK_MK_KN, SmallM)
{
std::vector<int> Ms{0, 1, 2, 3, 4, 5, 6};
int N = 512;
int K = 320;
constexpr int N = 512;
constexpr int K = 320;
int StrideA = K;
int StrideB = N;
int StrideC = N;
constexpr int StrideA = K;
constexpr int StrideB = N;
constexpr int StrideC = N;
for(int M : Ms)
this->Run(M, N, K, StrideA, StrideB, StrideC);
......@@ -17,12 +17,12 @@ TYPED_TEST(TestGemmSplitK_MK_KN, SmallM)
TYPED_TEST(TestGemmSplitK_MK_NK, SmallM)
{
std::vector<int> Ms{0, 1, 2, 3, 4, 5, 6};
int N = 512;
int K = 320;
constexpr int N = 512;
constexpr int K = 320;
int StrideA = K;
int StrideB = K;
int StrideC = N;
constexpr int StrideA = K;
constexpr int StrideB = K;
constexpr int StrideC = N;
for(int M : Ms)
this->Run(M, N, K, StrideA, StrideB, StrideC);
......@@ -31,11 +31,11 @@ TYPED_TEST(TestGemmSplitK_MK_NK, SmallM)
TYPED_TEST(TestGemmSplitK_KM_KN, SmallM)
{
std::vector<int> Ms{0, 1, 2, 3, 4, 5, 6};
int N = 512;
int K = 320;
constexpr int N = 512;
constexpr int K = 320;
int StrideB = N;
int StrideC = N;
constexpr int StrideB = N;
constexpr int StrideC = N;
for(int M : Ms)
this->Run(M, N, K, M, StrideB, StrideC);
......@@ -44,11 +44,11 @@ TYPED_TEST(TestGemmSplitK_KM_KN, SmallM)
TYPED_TEST(TestGemmSplitK_KM_NK, SmallM)
{
std::vector<int> Ms{0, 1, 2, 3, 4, 5, 6};
int N = 512;
int K = 320;
constexpr int N = 512;
constexpr int K = 320;
int StrideB = K;
int StrideC = N;
constexpr int StrideB = K;
constexpr int StrideC = N;
for(int M : Ms)
this->Run(M, N, K, M, StrideB, StrideC);
......@@ -57,12 +57,12 @@ TYPED_TEST(TestGemmSplitK_KM_NK, SmallM)
TYPED_TEST(TestGemmSplitK_MK_KN, MidLargeM)
{
std::vector<int> Ms{127, 255, 312, 799, 1573};
int N = 512;
int K = 320;
constexpr int N = 512;
constexpr int K = 320;
int StrideA = K;
int StrideB = N;
int StrideC = N;
constexpr int StrideA = K;
constexpr int StrideB = N;
constexpr int StrideC = N;
for(int M : Ms)
this->Run(M, N, K, StrideA, StrideB, StrideC);
......@@ -71,12 +71,12 @@ TYPED_TEST(TestGemmSplitK_MK_KN, MidLargeM)
TYPED_TEST(TestGemmSplitK_MK_NK, MidLargeM)
{
std::vector<int> Ms{127, 255, 312, 799, 1573};
int N = 512;
int K = 320;
constexpr int N = 512;
constexpr int K = 320;
int StrideA = K;
int StrideB = K;
int StrideC = N;
constexpr int StrideA = K;
constexpr int StrideB = K;
constexpr int StrideC = N;
for(int M : Ms)
this->Run(M, N, K, StrideA, StrideB, StrideC);
......@@ -85,11 +85,11 @@ TYPED_TEST(TestGemmSplitK_MK_NK, MidLargeM)
TYPED_TEST(TestGemmSplitK_KM_KN, MidLargeM)
{
std::vector<int> Ms{127, 255, 312, 799, 1573};
int N = 512;
int K = 320;
constexpr int N = 512;
constexpr int K = 320;
int StrideB = N;
int StrideC = N;
constexpr int StrideB = N;
constexpr int StrideC = N;
for(int M : Ms)
this->Run(M, N, K, M, StrideB, StrideC);
......@@ -98,11 +98,11 @@ TYPED_TEST(TestGemmSplitK_KM_KN, MidLargeM)
TYPED_TEST(TestGemmSplitK_KM_NK, MidLargeM)
{
std::vector<int> Ms{127, 255, 312, 799, 1573};
int N = 512;
int K = 320;
constexpr int N = 512;
constexpr int K = 320;
int StrideB = K;
int StrideC = N;
constexpr int StrideB = K;
constexpr int StrideC = N;
for(int M : Ms)
this->Run(M, N, K, M, StrideB, StrideC);
......@@ -111,12 +111,12 @@ TYPED_TEST(TestGemmSplitK_KM_NK, MidLargeM)
TYPED_TEST(TestGemmSplitK_MK_KN, PaddK)
{
std::vector<int> Ms{127};
int N = 512;
int K = 437;
constexpr int N = 512;
constexpr int K = 437;
int StrideA = K;
int StrideB = N;
int StrideC = N;
constexpr int StrideA = K;
constexpr int StrideB = N;
constexpr int StrideC = N;
for(int M : Ms)
this->Run(M, N, K, StrideA, StrideB, StrideC);
......@@ -125,12 +125,12 @@ TYPED_TEST(TestGemmSplitK_MK_KN, PaddK)
TYPED_TEST(TestGemmSplitK_MK_NK, PaddK)
{
std::vector<int> Ms{127};
int N = 512;
int K = 437;
constexpr int N = 512;
constexpr int K = 437;
int StrideA = K;
int StrideB = K;
int StrideC = N;
constexpr int StrideA = K;
constexpr int StrideB = K;
constexpr int StrideC = N;
for(int M : Ms)
this->Run(M, N, K, StrideA, StrideB, StrideC);
......@@ -139,11 +139,11 @@ TYPED_TEST(TestGemmSplitK_MK_NK, PaddK)
TYPED_TEST(TestGemmSplitK_KM_KN, PaddK)
{
std::vector<int> Ms{127};
int N = 512;
int K = 437;
constexpr int N = 512;
constexpr int K = 437;
int StrideB = N;
int StrideC = N;
constexpr int StrideB = N;
constexpr int StrideC = N;
for(int M : Ms)
this->Run(M, N, K, M, StrideB, StrideC);
......@@ -152,11 +152,11 @@ TYPED_TEST(TestGemmSplitK_KM_KN, PaddK)
TYPED_TEST(TestGemmSplitK_KM_NK, PaddK)
{
std::vector<int> Ms{127};
int N = 512;
int K = 437;
constexpr int N = 512;
constexpr int K = 437;
int StrideB = K;
int StrideC = N;
constexpr int StrideB = K;
constexpr int StrideC = N;
for(int M : Ms)
this->Run(M, N, K, M, StrideB, StrideC);
......@@ -165,12 +165,12 @@ TYPED_TEST(TestGemmSplitK_KM_NK, PaddK)
TYPED_TEST(TestGemmSplitK_MK_KN, Regular)
{
std::vector<int> Ms{512};
int N = 512;
int K = 512;
constexpr int N = 512;
constexpr int K = 512;
int StrideA = K;
int StrideB = N;
int StrideC = N;
constexpr int StrideA = K;
constexpr int StrideB = N;
constexpr int StrideC = N;
for(int M : Ms)
this->Run(M, N, K, StrideA, StrideB, StrideC);
......@@ -179,12 +179,12 @@ TYPED_TEST(TestGemmSplitK_MK_KN, Regular)
TYPED_TEST(TestGemmSplitK_MK_NK, Regular)
{
std::vector<int> Ms{512};
int N = 512;
int K = 512;
constexpr int N = 512;
constexpr int K = 512;
int StrideA = K;
int StrideB = K;
int StrideC = N;
constexpr int StrideA = K;
constexpr int StrideB = K;
constexpr int StrideC = N;
for(int M : Ms)
this->Run(M, N, K, StrideA, StrideB, StrideC);
......@@ -193,11 +193,11 @@ TYPED_TEST(TestGemmSplitK_MK_NK, Regular)
TYPED_TEST(TestGemmSplitK_KM_KN, Regular)
{
std::vector<int> Ms{512};
int N = 512;
int K = 512;
constexpr int N = 512;
constexpr int K = 512;
int StrideB = N;
int StrideC = N;
constexpr int StrideB = N;
constexpr int StrideC = N;
for(int M : Ms)
this->Run(M, N, K, M, StrideB, StrideC);
......@@ -206,11 +206,11 @@ TYPED_TEST(TestGemmSplitK_KM_KN, Regular)
TYPED_TEST(TestGemmSplitK_KM_NK, Regular)
{
std::vector<int> Ms{512};
int N = 512;
int K = 512;
constexpr int N = 512;
constexpr int K = 512;
int StrideB = K;
int StrideC = N;
constexpr int StrideB = K;
constexpr int StrideC = N;
for(int M : Ms)
this->Run(M, N, K, M, StrideB, StrideC);
......
test/gemm_split_k/test_gemm_splitk_util.hpp
View file @ 0c5f9438
......@@ -33,10 +33,10 @@ class TestGemmSplitK : public testing::Test
using CDataType = std::tuple_element_t<4, Tuple>;
public:
bool verify_ = true;
int init_method_ = 1; // decimal value initialization
bool log_ = false;
bool bench_ = false; // measure kernel performance
static constexpr bool verify_ = true;
static constexpr int init_method_ = 1; // decimal value initialization
static constexpr bool log_ = false;
static constexpr bool bench_ = false; // measure kernel performance
std::vector<int> k_batches_;
void SetUp() override { k_batches_ = {1, 2, 3, 5, 8}; }
......
test/grouped_gemm/test_grouped_gemm_interface.cpp
View file @ 0c5f9438
......@@ -45,8 +45,8 @@ class TestGGemmSplitKInterface_MKNKMN : public ::testing::Test
TEST_F(TestGGemmSplitKInterface_MKNKMN, TileSize)
{
std::vector<int> Ms{128, 256, 188, 512};
int N = 256;
int K = 128;
constexpr int N = 256;
constexpr int K = 128;
std::vector<int> Ns(Ms.size(), N);
std::vector<int> Ks(Ms.size(), K);
......@@ -70,8 +70,8 @@ TEST_F(TestGGemmSplitKInterface_MKNKMN, VectorLoadWidth)
using PaddedGGemmInstance = GGemmInstance<GemmMNKPadding, 32, 8, 4, 8, 8>;
std::vector<int> Ms{128, 256, 256, 512};
int N = 256;
int K = 512;
constexpr int N = 256;
constexpr int K = 512;
std::vector<int> Ns(Ms.size(), N);
std::vector<int> Ks(Ms.size(), K);
......@@ -96,9 +96,9 @@ TEST_F(TestGGemmSplitKInterface_MKNKMN, VectorLoadWidth)
TEST_F(TestGGemmSplitKInterface_MKNKMN, KLoops)
{
std::vector<int> Ms{128, 256, 256, 512};
int N = 256;
int K = 128;
int kbatch = 4;
constexpr int N = 256;
constexpr int K = 128;
constexpr int kbatch = 4;
std::vector<int> Ns(Ms.size(), N);
std::vector<int> Ks(Ms.size(), K);
......@@ -152,8 +152,8 @@ class TestGGemmSplitKInterface_KMKNNM : public ::testing::Test
TEST_F(TestGGemmSplitKInterface_KMKNNM, TileSize)
{
std::vector<int> Ms{128, 256, 188, 512};
int N = 256;
int K = 128;
constexpr int N = 256;
constexpr int K = 128;
std::vector<int> Ns(Ms.size(), N);
std::vector<int> Ks(Ms.size(), K);
......@@ -177,8 +177,8 @@ TEST_F(TestGGemmSplitKInterface_KMKNNM, VectorLoadWidth)
using PaddedGGemmInstance = GGemmInstance<GemmMNKPadding, 32, 8, 2, 8, 4>;
std::vector<int> Ms{128, 256, 256, 512};
int N = 256;
int K = 512;
constexpr int N = 256;
constexpr int K = 512;
std::vector<int> Ns(Ms.size(), N);
std::vector<int> Ks(Ms.size(), K);
......
test/grouped_gemm/test_grouped_gemm_ut_cases.inc
View file @ 0c5f9438
......@@ -3,8 +3,8 @@
TEST_P(RRR_F16_F16_F16, TinyCases)
{
const std::vector<int> Ms{0, 1};
const int N = 768;
const int K = 544;
constexpr int N = 768;
constexpr int K = 544;
const std::vector<int> Ns(Ms.size(), N);
const std::vector<int> Ks(Ms.size(), K);
......@@ -18,8 +18,8 @@ TEST_P(RRR_F16_F16_F16, TinyCases)
TEST_P(RRR_F16_F16_F16, SmallCases)
{
const std::vector<int> Ms{2, 1, 3, 4, 5, 0};
const int N = 768;
const int K = 544;
constexpr int N = 768;
constexpr int K = 544;
const std::vector<int> Ns(Ms.size(), N);
const std::vector<int> Ks(Ms.size(), K);
......@@ -33,8 +33,8 @@ TEST_P(RRR_F16_F16_F16, SmallCases)
TEST_P(RRR_F16_F16_F16, MidCases)
{
const std::vector<int> Ms{167, 183, 177, 153, 139, 204};
const int N = 768;
const int K = 544;
constexpr int N = 768;
constexpr int K = 544;
const std::vector<int> Ns(Ms.size(), N);
const std::vector<int> Ks(Ms.size(), K);
......@@ -48,8 +48,8 @@ TEST_P(RRR_F16_F16_F16, MidCases)
TEST_P(RRR_F16_F16_F16, Regular)
{
const std::vector<int> Ms{64, 128, 256};
const int N = 768;
const int K = 320;
constexpr int N = 768;
constexpr int K = 320;
const std::vector<int> Ns(Ms.size(), N);
const std::vector<int> Ks(Ms.size(), K);
......@@ -63,8 +63,8 @@ TEST_P(RRR_F16_F16_F16, Regular)
TEST_P(RRR_F16_F16_F16, MNKPadded)
{
const std::vector<int> Ms{127, 150, 188, 210};
const int N = 136;
const int K = 280;
constexpr int N = 136;
constexpr int K = 280;
const std::vector<int> Ns(Ms.size(), N);
const std::vector<int> Ks(Ms.size(), K);
......@@ -78,8 +78,8 @@ TEST_P(RRR_F16_F16_F16, MNKPadded)
TEST_P(RCR_F16_F16_F16, TinyCases)
{
const std::vector<int> Ms{0, 1};
const int N = 768;
const int K = 544;
constexpr int N = 768;
constexpr int K = 544;
const std::vector<int> Ns(Ms.size(), N);
const std::vector<int> Ks(Ms.size(), K);
......@@ -92,8 +92,8 @@ TEST_P(RCR_F16_F16_F16, TinyCases)
TEST_P(RCR_F16_F16_F16, SmallCases)
{
const std::vector<int> Ms{2, 1, 3, 4, 5, 0};
const int N = 768;
const int K = 544;
constexpr int N = 768;
constexpr int K = 544;
const std::vector<int> Ns(Ms.size(), N);
const std::vector<int> Ks(Ms.size(), K);
......@@ -107,8 +107,8 @@ TEST_P(RCR_F16_F16_F16, SmallCases)
TEST_P(RCR_F16_F16_F16, MidCases)
{
const std::vector<int> Ms{167, 183, 177, 153, 139, 204};
const int N = 768;
const int K = 544;
constexpr int N = 768;
constexpr int K = 544;
const std::vector<int> Ns(Ms.size(), N);
const std::vector<int> Ks(Ms.size(), K);
......@@ -122,8 +122,8 @@ TEST_P(RCR_F16_F16_F16, MidCases)
TEST_P(RCR_F16_F16_F16, Regular)
{
const std::vector<int> Ms{32, 64, 128, 256};
const int N = 768;
const int K = 320;
constexpr int N = 768;
constexpr int K = 320;
const std::vector<int> Ns(Ms.size(), N);
const std::vector<int> Ks(Ms.size(), K);
......@@ -137,8 +137,8 @@ TEST_P(RCR_F16_F16_F16, Regular)
TEST_P(RCR_F16_F16_F16, MNKPadded)
{
const std::vector<int> Ms{127, 150, 188, 210};
const int N = 136;
const int K = 280;
constexpr int N = 136;
constexpr int K = 280;
const std::vector<int> Ns(Ms.size(), N);
const std::vector<int> Ks(Ms.size(), K);
......@@ -152,8 +152,8 @@ TEST_P(RCR_F16_F16_F16, MNKPadded)
TEST_P(RRR_F16_F16_F16_LargeK, TestLargeKBatch)
{
const std::vector<int> Ms{188, 210};
const int N = 768;
const int K = 4096;
constexpr int N = 768;
constexpr int K = 4096;
const std::vector<int> Ns(Ms.size(), N);
const std::vector<int> Ks(Ms.size(), K);
......@@ -167,8 +167,8 @@ TEST_P(RRR_F16_F16_F16_LargeK, TestLargeKBatch)
TEST_P(RCR_F16_F16_F16_LargeK, TestLargeKBatch)
{
const std::vector<int> Ms{188, 210};
const int N = 768;
const int K = 4096;
constexpr int N = 768;
constexpr int K = 4096;
const std::vector<int> Ns(Ms.size(), N);
const std::vector<int> Ks(Ms.size(), K);
......
test/grouped_gemm/test_grouped_gemm_util.hpp
View file @ 0c5f9438
......@@ -50,10 +50,10 @@ class TestGroupedGemm : public testing::TestWithParam<int>
using EDataType = std::tuple_element_t<5, Tuple>;
public:
bool verify_ = true;
int init_method_ = 0; // decimal value initialization
bool log_ = false;
bool bench_ = false; // measure kernel performance
static constexpr bool verify_ = true;
static constexpr int init_method_ = 1; // decimal value initialization
static constexpr bool log_ = false;
static constexpr bool bench_ = false; // measure kernel performance
void SetUp() override {}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment