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
composable_kernel
Commits
e7a4548b
Commit
e7a4548b
authored
Jan 31, 2022
by
rocking
Browse files
Fix typo
parent
f1245c3b
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
4 deletions
+4
-4
composable_kernel/include/tensor_operation/element_wise_operation.hpp
...ernel/include/tensor_operation/element_wise_operation.hpp
+1
-1
example/2_gemm_xdl_bias/gemm_xdl_bias.cpp
example/2_gemm_xdl_bias/gemm_xdl_bias.cpp
+3
-3
No files found.
composable_kernel/include/tensor_operation/element_wise_operation.hpp
View file @
e7a4548b
...
@@ -28,7 +28,7 @@ struct Bias
...
@@ -28,7 +28,7 @@ struct Bias
{
{
__host__
__device__
constexpr
void
operator
()(
half_t
&
y_plus_bias
,
const
half_t
&
y
,
const
half_t
&
bias
)
const
__host__
__device__
constexpr
void
operator
()(
half_t
&
y_plus_bias
,
const
half_t
&
y
,
const
half_t
&
bias
)
const
{
{
y_plus_bias
=
y_plus_bias
=
y
+
bias
;
y_plus_bias
=
y
+
bias
;
}
}
};
};
...
...
example/2_gemm_xdl_bias/gemm_xdl_bias.cpp
View file @
e7a4548b
...
@@ -176,9 +176,9 @@ int main(int argc, char* argv[])
...
@@ -176,9 +176,9 @@ int main(int argc, char* argv[])
Tensor
<
ADataType
>
a_m_k
(
f_host_tensor_descriptor
(
M
,
K
,
StrideA
,
ALayout
{}));
Tensor
<
ADataType
>
a_m_k
(
f_host_tensor_descriptor
(
M
,
K
,
StrideA
,
ALayout
{}));
Tensor
<
BDataType
>
b_k_n
(
f_host_tensor_descriptor
(
K
,
N
,
StrideB
,
BLayout
{}));
Tensor
<
BDataType
>
b_k_n
(
f_host_tensor_descriptor
(
K
,
N
,
StrideB
,
BLayout
{}));
Tensor
<
B
DataType
>
c0_m_n
(
f_bias_host_tensor_descriptor
(
M
,
N
));
Tensor
<
C
DataType
>
c0_m_n
(
f_bias_host_tensor_descriptor
(
M
,
N
));
Tensor
<
B
DataType
>
c_m_n_host_result
(
f_host_tensor_descriptor
(
M
,
N
,
StrideC
,
CLayout
{}));
Tensor
<
C
DataType
>
c_m_n_host_result
(
f_host_tensor_descriptor
(
M
,
N
,
StrideC
,
CLayout
{}));
Tensor
<
B
DataType
>
c_m_n_device_result
(
f_host_tensor_descriptor
(
M
,
N
,
StrideC
,
CLayout
{}));
Tensor
<
C
DataType
>
c_m_n_device_result
(
f_host_tensor_descriptor
(
M
,
N
,
StrideC
,
CLayout
{}));
std
::
cout
<<
"a_m_k: "
<<
a_m_k
.
mDesc
<<
std
::
endl
;
std
::
cout
<<
"a_m_k: "
<<
a_m_k
.
mDesc
<<
std
::
endl
;
std
::
cout
<<
"b_k_n: "
<<
b_k_n
.
mDesc
<<
std
::
endl
;
std
::
cout
<<
"b_k_n: "
<<
b_k_n
.
mDesc
<<
std
::
endl
;
...
...
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