Commit ddd42c2d authored by Jakub Piasecki's avatar Jakub Piasecki
Browse files

added fp8 and bf8 instances

parent 478dfe12
...@@ -89,7 +89,7 @@ struct DataTypeTraits<ck_tile::bf8_t> ...@@ -89,7 +89,7 @@ struct DataTypeTraits<ck_tile::bf8_t>
static constexpr const char* name = "bf8"; static constexpr const char* name = "bf8";
}; };
/** \brief Struct used for specifying desired gemm details*/ /** @brief Struct used for specifying desired gemm details*/
struct gemm_traits struct gemm_traits
{ {
std::string data_type; /** Tensors datatype, can be set to either fp16 or bf16*/ std::string data_type; /** Tensors datatype, can be set to either fp16 or bf16*/
...@@ -146,12 +146,12 @@ template <typename Traits_> ...@@ -146,12 +146,12 @@ template <typename Traits_>
float gemm_(const ck_tile::GemmHostArgs& args, const ck_tile::stream_config& s); float gemm_(const ck_tile::GemmHostArgs& args, const ck_tile::stream_config& s);
/** /**
* \brief Invoke gemm function * @brief Invoke gemm function
* *
* \param traits Gemm traits which are used for choosing best instance. * @param traits Gemm traits which are used for choosing best instance.
* \param args Runtime gemm host arguments. * @param args Runtime gemm host arguments.
* \param s Stream configuration. * @param s Stream configuration.
* \return Time of execution. * @return Time of execution.
*/ */
float gemm(const gemm_traits& traits, float gemm(const gemm_traits& traits,
const ck_tile::GemmHostArgs& args, const ck_tile::GemmHostArgs& args,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment