Commit d97fb02a authored by Umang Yadav's avatar Umang Yadav
Browse files

merge changes

parent 6f768035
...@@ -69,8 +69,7 @@ struct ck_gemm ...@@ -69,8 +69,7 @@ struct ck_gemm
static bool is_ck_supported_type(shape::type_t t) static bool is_ck_supported_type(shape::type_t t)
{ {
return contains( return contains({shape::half_type, shape::int8_type, shape::int32_type}, t);
{shape::half_type, shape::int8_type, shape::int32_type, shape::fp8e4m3fnuz_type}, t);
} }
}; };
MIGRAPHX_REGISTER_OP(ck_gemm); MIGRAPHX_REGISTER_OP(ck_gemm);
......
...@@ -97,7 +97,6 @@ std::vector<pass> target::get_passes(migraphx::context& gctx, const compile_opti ...@@ -97,7 +97,6 @@ std::vector<pass> target::get_passes(migraphx::context& gctx, const compile_opti
auto& ctx = any_cast<context>(gctx); auto& ctx = any_cast<context>(gctx);
ctx.set_exhaustive_tune_flag(options.exhaustive_tune); ctx.set_exhaustive_tune_flag(options.exhaustive_tune);
std::set<shape::type_t> unsupported_types(shape::types().begin(), shape::types().end()); std::set<shape::type_t> unsupported_types(shape::types().begin(), shape::types().end());
unsupported_types.erase(shape::type_t::fp8e4m3fnuz_type);
unsupported_types.erase(shape::type_t::float_type); unsupported_types.erase(shape::type_t::float_type);
unsupported_types.erase(shape::type_t::fp8e4m3fnuz_type); unsupported_types.erase(shape::type_t::fp8e4m3fnuz_type);
unsupported_types.erase(shape::type_t::half_type); unsupported_types.erase(shape::type_t::half_type);
......
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