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
8ba66ebc
Commit
8ba66ebc
authored
Nov 10, 2023
by
Paul
Browse files
Format
parent
7fa93728
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
4 deletions
+4
-4
src/targets/gpu/kernels/include/migraphx/kernels/concat.hpp
src/targets/gpu/kernels/include/migraphx/kernels/concat.hpp
+4
-4
No files found.
src/targets/gpu/kernels/include/migraphx/kernels/concat.hpp
View file @
8ba66ebc
...
@@ -81,15 +81,15 @@ __device__ auto concat2(InputPacks... input_packs)
...
@@ -81,15 +81,15 @@ __device__ auto concat2(InputPacks... input_packs)
auto
idx
=
make_index
();
auto
idx
=
make_index
();
fold
([
&
](
auto
start
,
auto
input_pack
)
{
fold
([
&
](
auto
start
,
auto
input_pack
)
{
return
input_pack
([
&
](
auto
g
,
auto
x
,
auto
...
xs
)
{
return
input_pack
([
&
](
auto
g
,
auto
x
,
auto
...
xs
)
{
concat_slices
<
Axis
>
(
x
,
start
,
ts
...)([
&
](
auto
z
,
auto
...
ys
)
{
concat_slices
<
Axis
>
(
x
,
start
,
ts
...)([
&
](
auto
z
,
auto
...
ys
)
{
idx
.
global_stride
(
x
.
get_shape
().
elements
(),
idx
.
global_stride
(
x
.
get_shape
().
elements
(),
[
&
](
auto
i
)
{
z
[
i
]
=
f
(
g
(
x
[
i
],
xs
[
i
]...),
ys
[
i
]...);
});
[
&
](
auto
i
)
{
z
[
i
]
=
f
(
g
(
x
[
i
],
xs
[
i
]...),
ys
[
i
]...);
});
return
start
+
concat_ends
<
Axis
>
(
x
);
return
start
+
concat_ends
<
Axis
>
(
x
);
});
});
});
});
})(
_c
<
0
>
,
input_packs
...);
})(
_c
<
0
>
,
input_packs
...);
}
}
}
// namespace migraphx
}
// namespace migraphx
#endif // MIGRAPHX_GUARD_KERNELS_CONCAT_HPP
#endif // MIGRAPHX_GUARD_KERNELS_CONCAT_HPP
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