Commit e977ab07 authored by Paul's avatar Paul
Browse files

Use another ifdef

parent fd5a68a7
...@@ -12,6 +12,7 @@ struct module; ...@@ -12,6 +12,7 @@ struct module;
namespace gpu { namespace gpu {
#ifdef MIGRAPHX_MLIR
struct mlir_conv struct mlir_conv
{ {
operation op = make_op("convolution"); operation op = make_op("convolution");
...@@ -40,7 +41,7 @@ namespace { ...@@ -40,7 +41,7 @@ namespace {
struct find_conv_pointwise struct find_conv_pointwise
{ {
// Find a convolution followed by a pointwise operation. // Find a convolution followed by a pointwise operation.
auto matcher() const // NOLINT auto matcher() const
{ {
auto convolution = auto convolution =
match::skip(match::name("contiguous"))(match::name("convolution").bind("convolution")); match::skip(match::name("contiguous"))(match::name("convolution").bind("convolution"));
...@@ -98,6 +99,8 @@ struct find_conv_pointwise ...@@ -98,6 +99,8 @@ struct find_conv_pointwise
}; };
} // namespace } // namespace
#endif
void fuse_mlir::apply(module_pass_manager& mpm) const void fuse_mlir::apply(module_pass_manager& mpm) const
{ {
#ifdef MIGRAPHX_MLIR #ifdef MIGRAPHX_MLIR
......
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