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
db2def39
"mmdet3d/datasets/vscode:/vscode.git/clone" did not exist on "c2c0f3d80c15abd42be973294695a04d9679ada1"
Commit
db2def39
authored
May 10, 2022
by
Paul
Browse files
Format
parent
f1f60be1
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
9 additions
and
12 deletions
+9
-12
src/targets/gpu/compile_gen.cpp
src/targets/gpu/compile_gen.cpp
+2
-2
src/targets/gpu/jit/reduce.cpp
src/targets/gpu/jit/reduce.cpp
+7
-10
No files found.
src/targets/gpu/compile_gen.cpp
View file @
db2def39
...
@@ -30,7 +30,7 @@ vectorize vectorize::elements(std::size_t axis, const std::vector<shape>& inputs
...
@@ -30,7 +30,7 @@ vectorize vectorize::elements(std::size_t axis, const std::vector<shape>& inputs
auto
len
=
input
.
lens
()[
axis
];
auto
len
=
input
.
lens
()[
axis
];
if
(
stride
!=
0
and
stride
!=
1
)
if
(
stride
!=
0
and
stride
!=
1
)
return
1
;
return
1
;
if
(
len
==
1
)
if
(
len
==
1
)
return
sizes
.
front
();
return
sizes
.
front
();
auto
it
=
std
::
find_if
(
auto
it
=
std
::
find_if
(
sizes
.
begin
(),
sizes
.
end
(),
[
&
](
auto
i
)
{
return
(
len
%
i
)
==
0
;
});
sizes
.
begin
(),
sizes
.
end
(),
[
&
](
auto
i
)
{
return
(
len
%
i
)
==
0
;
});
...
...
src/targets/gpu/jit/reduce.cpp
View file @
db2def39
...
@@ -115,14 +115,11 @@ struct reduce_compiler : compiler<reduce_compiler>
...
@@ -115,14 +115,11 @@ struct reduce_compiler : compiler<reduce_compiler>
{
{
auto
block_size
=
compute_block_size
(
relements
,
256
);
auto
block_size
=
compute_block_size
(
relements
,
256
);
options
.
set_launch_params
(
options
.
set_launch_params
(
v
,
v
,
compute_global_for
(
ctx
,
nelements
*
block_size
,
256
),
block_size
);
compute_global_for
(
ctx
,
nelements
*
block_size
,
256
),
block_size
);
}
}
else
if
(
algo
==
"lane"
)
else
if
(
algo
==
"lane"
)
{
{
options
.
set_launch_params
(
options
.
set_launch_params
(
v
,
compute_global_for
(
ctx
,
nelements
,
256
));
v
,
compute_global_for
(
ctx
,
nelements
,
256
));
}
}
else
else
{
{
...
...
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