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
OpenDAS
TransformerEngine
Commits
c636071d
Commit
c636071d
authored
May 22, 2025
by
wenjh
Browse files
Fix build error of userbuffer.cu
Signed-off-by:
wenjh
<
wenjh@sugon.com
>
parent
6ed9a3e4
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
transformer_engine/common/comm_gemm_overlap/userbuffers/userbuffers.cu
...ngine/common/comm_gemm_overlap/userbuffers/userbuffers.cu
+2
-2
No files found.
transformer_engine/common/comm_gemm_overlap/userbuffers/userbuffers.cu
View file @
c636071d
...
...
@@ -2067,7 +2067,7 @@ template void reducescatter2_userbuff_strided_atomic_fp8<te_hip_fp8_e4m3>(
void
*
output
,
float
*
scale
,
const
int
handler
,
const
int
offset
,
const
int
rowelements
,
const
int
colelements
,
const
int
strideelements_out
,
const
int
strideelements_in
,
const
int
numchunks
,
void
*
counters
,
communicator
*
comm
,
cudaStream_t
stream
);
template
void
reducescatter2_userbuff_strided_atomic_fp8
<
hip_f8
<
te_hip_fp8_e5m2
>(
template
void
reducescatter2_userbuff_strided_atomic_fp8
<
te_hip_fp8_e5m2
>(
void
*
output
,
float
*
scale
,
const
int
handler
,
const
int
offset
,
const
int
rowelements
,
const
int
colelements
,
const
int
strideelements_out
,
const
int
strideelements_in
,
const
int
numchunks
,
void
*
counters
,
communicator
*
comm
,
cudaStream_t
stream
);
...
...
@@ -2075,7 +2075,7 @@ template void reducescatter2_userbuff_strided_multiatomic_fp8<te_hip_fp8_e4m3>(
void
*
output
,
float
*
scale
,
const
int
handler
,
const
int
offset
,
const
int
rowelements
,
const
int
colelements
,
const
int
strideelements_out
,
const
int
strideelements_in
,
const
int
numchunks
,
void
*
counters
,
communicator
*
comm
,
cudaStream_t
stream
);
template
void
reducescatter2_userbuff_strided_multiatomic_fp8
<
hip_f8
<
te_hip_fp8_e5m2
>(
template
void
reducescatter2_userbuff_strided_multiatomic_fp8
<
te_hip_fp8_e5m2
>(
void
*
output
,
float
*
scale
,
const
int
handler
,
const
int
offset
,
const
int
rowelements
,
const
int
colelements
,
const
int
strideelements_out
,
const
int
strideelements_in
,
const
int
numchunks
,
void
*
counters
,
communicator
*
comm
,
cudaStream_t
stream
);
...
...
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