Skip to content

Commit

Permalink
Fix Windows failure
Browse files Browse the repository at this point in the history
  • Loading branch information
ahsan-ca committed Dec 12, 2024
1 parent 733d670 commit 3926315
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions src/targets/gpu/fuse_ops.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -814,6 +814,7 @@ struct find_contiguous_transpose_rocblas_gemm : contiguous_transpose_gemm
}
};

#if MIGRAPHX_USE_HIPBLASLT
struct find_contiguous_transpose_hip_gemm : contiguous_transpose_gemm
{
auto matcher() const
Expand Down Expand Up @@ -872,6 +873,7 @@ struct find_contiguous_transpose_hip_gemm : contiguous_transpose_gemm
m.replace_instruction(ins, gemm_transpoe);
}
};
#endif

struct find_commutative_broadcast
{
Expand Down Expand Up @@ -1043,7 +1045,9 @@ void fuse_ops::apply(module& m) const
find_layernorm_pointwise{},
find_concat_pointwise{},
find_contiguous_transpose_rocblas_gemm{},
#if MIGRAPHX_USE_HIPBLASLT
find_contiguous_transpose_hip_gemm{},
#endif
find_commutative_broadcast{});
match::find_matches(m, find_contiguous{});
}
Expand Down

0 comments on commit 3926315

Please sign in to comment.