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
84e198a3
Commit
84e198a3
authored
Jun 09, 2025
by
wenjh
Browse files
Merge branch 'develop_v2.4'
parents
638296df
7d2b9c77
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
tests/cpp/operator/test_cublaslt_gemm.cu
tests/cpp/operator/test_cublaslt_gemm.cu
+2
-2
No files found.
tests/cpp/operator/test_cublaslt_gemm.cu
View file @
84e198a3
...
@@ -116,11 +116,11 @@ void performTest(bool use_bias, bool use_gelu, const size_t m, const size_t k, c
...
@@ -116,11 +116,11 @@ void performTest(bool use_bias, bool use_gelu, const size_t m, const size_t k, c
Tensor
D
(
"D"
,
std
::
vector
<
size_t
>
{
n
,
m
},
dtype
);
Tensor
D
(
"D"
,
std
::
vector
<
size_t
>
{
n
,
m
},
dtype
);
Tensor
bias
;
Tensor
bias
;
if
(
use_bias
){
if
(
use_bias
){
bias
=
Tensor
(
"bias"
,
{
m
},
bias_type
);
bias
=
Tensor
(
"bias"
,
std
::
vector
<
size_t
>
{
m
},
bias_type
);
}
}
Tensor
pre_gelu_out
;
Tensor
pre_gelu_out
;
if
(
use_gelu
){
if
(
use_gelu
){
pre_gelu_out
=
Tensor
(
"pre_gelu_out"
,
{
n
,
m
},
gelu_type
);
pre_gelu_out
=
Tensor
(
"pre_gelu_out"
,
std
::
vector
<
size_t
>
{
n
,
m
},
gelu_type
);
}
}
//initialize the data and scale inv of A, B
//initialize the data and scale inv of A, B
...
...
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