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
63330bff
Commit
63330bff
authored
Dec 12, 2023
by
Paul
Browse files
Rename to just readlane
parent
282fd136
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
2 deletions
+2
-2
src/targets/gpu/kernels/include/migraphx/kernels/dpp.hpp
src/targets/gpu/kernels/include/migraphx/kernels/dpp.hpp
+1
-1
src/targets/gpu/kernels/include/migraphx/kernels/reduce.hpp
src/targets/gpu/kernels/include/migraphx/kernels/reduce.hpp
+1
-1
No files found.
src/targets/gpu/kernels/include/migraphx/kernels/dpp.hpp
View file @
63330bff
...
...
@@ -89,7 +89,7 @@ __device__ T dpp_swizzle(T& x)
}
template
<
unsigned
int
SrcLane
,
unsigned
int
Width
,
class
T
>
__device__
T
dpp_
readlane
(
T
&
x
)
__device__
T
readlane
(
T
&
x
)
{
static_assert
(
is_power_of_2
(
Width
),
"Width must be a power of 2"
);
return
dpp_op
(
x
,
[](
auto
i
)
{
return
__shfl
(
i
,
SrcLane
,
Width
);
});
...
...
src/targets/gpu/kernels/include/migraphx/kernels/reduce.hpp
View file @
63330bff
...
...
@@ -172,7 +172,7 @@ __device__ auto subwave_reduce(index idx, Op op, T init, Index n, F f)
idx
.
local_subwave_stride
<
SubWaveSize
>
(
n
,
[
&
](
auto
i
,
auto
d
)
{
x
=
op
(
x
,
index
::
invoke_loop
(
f
,
i
,
d
));
});
dpp_reduce
<
SubWaveSize
>
(
x
,
op
);
return
dpp_
readlane
<
SubWaveSize
-
1
,
SubWaveSize
>
(
x
);
return
readlane
<
SubWaveSize
-
1
,
SubWaveSize
>
(
x
);
}
template
<
class
Op
,
class
T
,
class
Index
,
class
F
>
...
...
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