ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 22:39:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix build breaks in the elementwise examples (#2789)

Browse Source 
* Include variant in elementwise_common.hpp

* Disallow bf16_t for both UnarySquare and UnaryConvert in elementwise_example_unary.cpp
This commit is contained in:
SamiAario-AMD
2025-09-05 19:48:01 +03:00
committed by GitHub
parent b740380906
commit 7d81f85d15
2 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
example/ck_tile/21_elementwise/elementwise_common.hpp
View File

@@ -1,6 +1,7 @@
// SPDX-License-Identifier: MIT
// Copyright (c) 2025, Advanced Micro Devices, Inc. All rights reserved.
#include <variant>
#include "ck_tile/core/arch/arch.hpp"
auto string_to_datatype(const std::string& datatype)

9
example/ck_tile/21_elementwise/elementwise_example_unary.cpp
View File

@@ -158,7 +158,14 @@ bool filter_then_run(const ck_tile::ArgParser& arg_parser)
bool pass = true;
if constexpr(std::is_same_v<XElementwiseOperation, ck_tile::element_wise::UnarySquare> &&
std::is_same_v<XDataType, ck_tile::bf16_t>)
(std::is_same_v<XDataType, ck_tile::bf16_t> ||
std::is_same_v<YDataType, ck_tile::bf16_t>))
{
throw_unsupported();
}
else if constexpr(std::is_same_v<XElementwiseOperation, ck_tile::element_wise::UnaryConvert> &&
(std::is_same_v<XDataType, ck_tile::bf16_t> ||
std::is_same_v<YDataType, ck_tile::bf16_t>))
{
throw_unsupported();
}