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
jerrrrry
infinicore
Commits
53468445
Commit
53468445
authored
Jun 26, 2025
by
Catheriany
Browse files
issue/282: 算子贴错导致推理问题修复
parent
af8bdb43
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
src/infiniop/ops/causal_softmax/maca/causal_softmax_kernel.h
src/infiniop/ops/causal_softmax/maca/causal_softmax_kernel.h
+1
-1
No files found.
src/infiniop/ops/causal_softmax/maca/causal_softmax_kernel.h
View file @
53468445
...
@@ -18,7 +18,7 @@ INFINIOP_MACA_KERNEL causalSoftmax(
...
@@ -18,7 +18,7 @@ INFINIOP_MACA_KERNEL causalSoftmax(
// [Reduce] Find max value in each row and store in shared memory
// [Reduce] Find max value in each row and store in shared memory
__shared__
Tdata
max_
;
__shared__
Tdata
max_
;
Tdata
max_0
=
op
::
common_maca
::
reduce_op
::
max
<
BLOCK_SIZE
,
Tdata
>
(
x
,
width
);
Tdata
max_0
=
op
::
common_maca
::
reduce_op
::
max
<
BLOCK_SIZE
,
Tdata
>
(
x
,
width
-
height
+
1
+
blockIdx
.
x
);
if
(
threadIdx
.
x
==
0
)
{
if
(
threadIdx
.
x
==
0
)
{
max_
=
max_0
;
max_
=
max_0
;
}
}
...
...
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