mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-03-25 01:27:40 +00:00
d85f065b15
* chore(copyright): update copyright header for codegen directory * chore(copyright): update copyright header for example directory
28 lines
646 B
C++
28 lines
646 B
C++
// Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
#include <variant>
|
|
#include "ck_tile/core/arch/arch.hpp"
|
|
|
|
auto string_to_datatype(const std::string& datatype)
|
|
{
|
|
using PrecVariant = std::variant<ck_tile::half_t, ck_tile::bf16_t, float>;
|
|
|
|
if(datatype == "fp16")
|
|
{
|
|
return PrecVariant{ck_tile::half_t{}};
|
|
}
|
|
else if(datatype == "bf16")
|
|
{
|
|
return PrecVariant{ck_tile::bf16_t{}};
|
|
}
|
|
else if(datatype == "fp32")
|
|
{
|
|
return PrecVariant{float{}};
|
|
}
|
|
else
|
|
{
|
|
throw std::runtime_error("Unsupported data type: " + datatype);
|
|
}
|
|
};
|