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
fdd29d75
Commit
fdd29d75
authored
Nov 16, 2023
by
Paul
Browse files
Keep MIGRAPHX_ENABLE_EXTRA_MLIR
parent
6f3784aa
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
0 deletions
+3
-0
src/targets/gpu/fuse_mlir.cpp
src/targets/gpu/fuse_mlir.cpp
+3
-0
No files found.
src/targets/gpu/fuse_mlir.cpp
View file @
fdd29d75
...
...
@@ -36,6 +36,7 @@ struct module;
namespace
gpu
{
MIGRAPHX_DECLARE_ENV_VAR
(
MIGRAPHX_ENABLE_EXTRA_MLIR
);
MIGRAPHX_DECLARE_ENV_VAR
(
MIGRAPHX_DISABLE_MLIR
);
bool
mlir_enabled
()
...
...
@@ -171,6 +172,8 @@ auto is_mlir_dot(mlir_mode mode)
float
g
=
a
.
elements
()
/
(
m
*
k
);
if
(
k
>
1024
)
return
false
;
if
(
enabled
(
MIGRAPHX_ENABLE_EXTRA_MLIR
{}))
return
true
;
auto
ratio
=
std
::
sqrt
(
g
)
*
m
*
n
/
k
;
if
(
ratio
<
2048
)
return
false
;
...
...
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