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
891be203
"official/modeling/optimization/legacy_adamw.py" did not exist on "1fbb7a82d552c02fc1803c92d68a81e85b42ccc8"
Commit
891be203
authored
May 25, 2023
by
Paul
Browse files
Add group stride
parent
b8b2e8ab
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
8 additions
and
0 deletions
+8
-0
src/targets/gpu/kernels/include/migraphx/kernels/index.hpp
src/targets/gpu/kernels/include/migraphx/kernels/index.hpp
+8
-0
No files found.
src/targets/gpu/kernels/include/migraphx/kernels/index.hpp
View file @
891be203
...
...
@@ -130,6 +130,8 @@ struct index
return
blockDim
.
x
;
}
#endif
constexpr
auto
ngroup
()
const
{
return
nglobal
()
/
max_nlocal
();
}
template
<
class
N
,
class
Stride
>
static
constexpr
auto
max_stride_iterations
(
N
n
,
Stride
stride
)
{
...
...
@@ -231,6 +233,12 @@ struct index
{
for_stride
<
true
>
(
local
,
n
,
nlocal
(),
f
);
}
template
<
class
F
,
class
N
>
__device__
void
group_stride
(
N
n
,
F
f
)
const
{
for_stride
<
false
>
(
group
,
n
,
ngroup
(),
f
);
}
};
#ifdef MIGRAPHX_NLOCAL
...
...
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