mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-16 02:54:21 +00:00
Merge commit '280bc4219151c3f79fe8ca076a2d10df4ff88b34' into develop
This commit is contained in:
35
experimental/builder/test/unit_conv_thread_block.cpp
Normal file
35
experimental/builder/test/unit_conv_thread_block.cpp
Normal file
@@ -0,0 +1,35 @@
|
||||
// Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
|
||||
// SPDX-License-Identifier: MIT
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include "ck_tile/builder/factory/helpers/conv_thread_block.hpp"
|
||||
|
||||
namespace {
|
||||
|
||||
using ::ck_tile::builder::factory::internal::ConvBlock;
|
||||
using ::ck_tile::builder::factory::internal::SetThreadBlockInfo;
|
||||
|
||||
TEST(ConvThreadBlock, AssignsThreadBlockAndTileSize)
|
||||
{
|
||||
constexpr struct Algorithm
|
||||
{
|
||||
struct ThreadBlock
|
||||
{
|
||||
int block_size = 256;
|
||||
struct TileSize
|
||||
{
|
||||
int m = 128;
|
||||
int n = 128;
|
||||
int k = 16;
|
||||
} tile_size;
|
||||
} thread_block;
|
||||
} kAlgorithm;
|
||||
constexpr ConvBlock block_info = SetThreadBlockInfo<kAlgorithm>();
|
||||
|
||||
EXPECT_EQ(block_info.block_size, 256);
|
||||
EXPECT_EQ(block_info.per_block.m, 128);
|
||||
EXPECT_EQ(block_info.per_block.n, 128);
|
||||
EXPECT_EQ(block_info.per_block.k, 16);
|
||||
}
|
||||
|
||||
} // namespace
|
||||
Reference in New Issue
Block a user