Fix build breaks in the elementwise examples (#2789)

* 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

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();
}