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

[ROCm/composable_kernel commit: 7d81f85d15]
This commit is contained in:
SamiAario-AMD
2025-09-05 19:48:01 +03:00
committed by GitHub
parent 48db870e4d
commit 43eb264e14
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();
}