Skip to content

GitLab

Commit 87548b5d authored by Umang Yadav's avatar Umang Yadav
Browse files

add ifdef to compile

parent 037205c5
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/targets/gpu/gemm_impl.cpp
View file @ 87548b5d
...@@ -219,6 +219,7 @@ struct gemm_impl ...@@ -219,6 +219,7 @@ struct gemm_impl
void run(context& ctx, const std::vector<argument>& input_args, int32_t solution_idx = 0) const void run(context& ctx, const std::vector<argument>& input_args, int32_t solution_idx = 0) const
{ {
#ifdef MIGRAPHX_USE_ROCBLAS_FP8_API
if(rocblas_fp8_available() and if(rocblas_fp8_available() and
std::any_of(input_args.begin(), input_args.end(), [](const auto i) { std::any_of(input_args.begin(), input_args.end(), [](const auto i) {
return i.get_shape().type() == migraphx::shape::fp8e4m3fnuz_type; return i.get_shape().type() == migraphx::shape::fp8e4m3fnuz_type;
...@@ -244,6 +245,7 @@ struct gemm_impl ...@@ -244,6 +245,7 @@ struct gemm_impl
} }
} }
else else
#endif
{ {
if(strided_batched) if(strided_batched)
{ {
......
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