Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
MIGraphX
Commits
6f3784aa
Commit
6f3784aa
authored
Nov 16, 2023
by
Paul
Browse files
Cleanup
parent
57b2ee1a
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
0 additions
and
9 deletions
+0
-9
src/targets/gpu/fuse_mlir.cpp
src/targets/gpu/fuse_mlir.cpp
+0
-9
No files found.
src/targets/gpu/fuse_mlir.cpp
View file @
6f3784aa
...
@@ -36,7 +36,6 @@ struct module;
...
@@ -36,7 +36,6 @@ struct module;
namespace
gpu
{
namespace
gpu
{
MIGRAPHX_DECLARE_ENV_VAR
(
MIGRAPHX_ENABLE_EXTRA_MLIR
);
MIGRAPHX_DECLARE_ENV_VAR
(
MIGRAPHX_DISABLE_MLIR
);
MIGRAPHX_DECLARE_ENV_VAR
(
MIGRAPHX_DISABLE_MLIR
);
bool
mlir_enabled
()
bool
mlir_enabled
()
...
@@ -173,14 +172,9 @@ auto is_mlir_dot(mlir_mode mode)
...
@@ -173,14 +172,9 @@ auto is_mlir_dot(mlir_mode mode)
if
(
k
>
1024
)
if
(
k
>
1024
)
return
false
;
return
false
;
auto
ratio
=
std
::
sqrt
(
g
)
*
m
*
n
/
k
;
auto
ratio
=
std
::
sqrt
(
g
)
*
m
*
n
/
k
;
std
::
cout
<<
"gemm_ratio: "
<<
ratio
<<
std
::
endl
;
if
(
ratio
<
2048
)
if
(
ratio
<
2048
)
return
false
;
return
false
;
return
true
;
return
true
;
// // Skipping GEMMs with a K dimension greater than 2048 is a course-grained strategy
// // to avoid poor-performing GEMM kernels from MLIR
// // To-do: Investigate a more precise strategy
// return k <= 2048;
});
});
}
}
...
@@ -421,9 +415,6 @@ void fuse_mlir::apply(module_pass_manager& mpm) const
...
@@ -421,9 +415,6 @@ void fuse_mlir::apply(module_pass_manager& mpm) const
return
std
::
max
(
m1
,
m2
);
return
std
::
max
(
m1
,
m2
);
};
};
mlir_mode
mode
=
(
enabled
(
MIGRAPHX_ENABLE_EXTRA_MLIR
{})
or
enable_extra
)
?
mlir_mode
::
fast
:
mlir_mode
::
none
;
match
::
find_matches
(
mpm
,
match
::
find_matches
(
mpm
,
find_mlir_fused_ops
{.
conv_mode
=
get_mode
(
"fused"
,
mlir_mode
::
fast
),
find_mlir_fused_ops
{.
conv_mode
=
get_mode
(
"fused"
,
mlir_mode
::
fast
),
.
dot_mode
=
get_mode
(
"fused"
,
mlir_mode
::
fast
)});
.
dot_mode
=
get_mode
(
"fused"
,
mlir_mode
::
fast
)});
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment