Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
DeepEP
Commits
77ddb015
"ssh:/git@developer.sourcefind.cn:2222/OpenDAS/DeepEP.git" did not exist on "243eca854ff763bc2b10648849a3fdf78edb9eda"
Commit
77ddb015
authored
Jul 02, 2025
by
Chenggang Zhao
Browse files
Remove launch bound for layout kernels
parent
d4f34978
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
5 deletions
+4
-5
csrc/kernels/layout.cu
csrc/kernels/layout.cu
+4
-5
No files found.
csrc/kernels/layout.cu
View file @
77ddb015
...
@@ -7,11 +7,10 @@ namespace deep_ep {
...
@@ -7,11 +7,10 @@ namespace deep_ep {
namespace
layout
{
namespace
layout
{
template
<
int
kNumThreads
,
int
kNumExpertsPerSM
,
int
kNumRanksPerSM
>
template
<
int
kNumThreads
,
int
kNumExpertsPerSM
,
int
kNumRanksPerSM
>
__global__
void
__launch_bounds__
(
kNumThreads
,
1
)
__global__
void
get_dispatch_layout
(
const
int64_t
*
topk_idx
,
get_dispatch_layout
(
const
int64_t
*
topk_idx
,
int
*
num_tokens_per_rank
,
int
*
num_tokens_per_rdma_rank
,
int
*
num_tokens_per_rank
,
int
*
num_tokens_per_rdma_rank
,
int
*
num_tokens_per_expert
,
bool
*
is_token_in_rank
,
int
*
num_tokens_per_expert
,
bool
*
is_token_in_rank
,
int
num_tokens
,
int
num_topk
,
int
num_ranks
,
int
num_experts
)
{
int
num_tokens
,
int
num_topk
,
int
num_ranks
,
int
num_experts
)
{
auto
sm_id
=
static_cast
<
int
>
(
blockIdx
.
x
);
auto
sm_id
=
static_cast
<
int
>
(
blockIdx
.
x
);
auto
thread_id
=
static_cast
<
int
>
(
threadIdx
.
x
);
auto
thread_id
=
static_cast
<
int
>
(
threadIdx
.
x
);
...
...
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