mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-02 12:41:26 +00:00
5cb8109535
Fix the Composable Kernel CI and versions incompatibility (#4640) ## Motivation This PR has 4 patches: 1. Fix the CI error of grouped gemm. 2. Fix the incompatibility of old linux version. 3. Fix the potential errors of flatmm. 4. Address the previous comments of abquant eight warps pipeline solution.
46 lines
2.3 KiB
C++
46 lines
2.3 KiB
C++
// Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
#include <tuple>
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "ck_tile/host.hpp"
|
|
#include "test_grouped_gemm_util.hpp"
|
|
|
|
using F16 = ck_tile::half_t;
|
|
using F32 = float;
|
|
using BF16 = ck_tile::bf16_t;
|
|
using Row = ck_tile::tensor_layout::gemm::RowMajor;
|
|
using Col = ck_tile::tensor_layout::gemm::ColumnMajor;
|
|
using True = ck_tile::bool_constant<true>;
|
|
using False = ck_tile::bool_constant<false>;
|
|
|
|
// clang-format off
|
|
using KernelTypes = ::testing::Types<
|
|
// ALayout, BLayout, CLayout, ADataType, BDataType, AccDataType, CDataType, Persistent
|
|
std::tuple< Row, Col, Row, F16, F16, F32, F16, True>,
|
|
std::tuple< Row, Col, Row, F16, F16, F32, F16, False>,
|
|
std::tuple< Row, Col, Row, F16, F16, F32, F16, True>,
|
|
std::tuple< Row, Col, Row, F16, F16, F32, F16, False>,
|
|
|
|
std::tuple< Col, Col, Row, F16, F16, F32, F16, True>,
|
|
std::tuple< Col, Col, Row, F16, F16, F32, F16, False>,
|
|
std::tuple< Row, Row, Row, F16, F16, F32, F16, True>,
|
|
std::tuple< Row, Row, Row, F16, F16, F32, F16, False>,
|
|
std::tuple< Col, Row, Row, F16, F16, F32, F16, True>,
|
|
std::tuple< Col, Row, Row, F16, F16, F32, F16, False>,
|
|
|
|
std::tuple< Col, Col, Row, BF16, BF16, F32, BF16, True>,
|
|
std::tuple< Col, Col, Row, BF16, BF16, F32, BF16, False>,
|
|
std::tuple< Row, Row, Row, BF16, BF16, F32, BF16, True>,
|
|
std::tuple< Row, Row, Row, BF16, BF16, F32, BF16, False>,
|
|
std::tuple< Col, Row, Row, BF16, BF16, F32, BF16, True>,
|
|
std::tuple< Col, Row, Row, BF16, BF16, F32, BF16, False>
|
|
>;
|
|
// clang-format on
|
|
|
|
TYPED_TEST_SUITE(TestCkTileGroupedGemm, KernelTypes);
|
|
|
|
#include "test_grouped_gemm_ut_cases.inc"
|