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
94d5f723
"...composable_kernel_rocm.git" did not exist on "bdcd037428ac356e5b77271b7b6669c5c2d9548a"
Commit
94d5f723
authored
Jun 14, 2022
by
rocking
Browse files
Fix static_cast order
parent
3d091db2
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
4 deletions
+4
-4
example/21_gemm_layernorm/gemm_bias_relu_add_layernorm_xdl_fp16.cpp
..._gemm_layernorm/gemm_bias_relu_add_layernorm_xdl_fp16.cpp
+2
-2
profiler/include/profile_gemm_bias_add_reduce_impl.hpp
profiler/include/profile_gemm_bias_add_reduce_impl.hpp
+2
-2
No files found.
example/21_gemm_layernorm/gemm_bias_relu_add_layernorm_xdl_fp16.cpp
View file @
94d5f723
...
@@ -169,11 +169,11 @@ void host_gemm_layernorm(Tensor<LayerNormOutDataType>& out_m_n,
...
@@ -169,11 +169,11 @@ void host_gemm_layernorm(Tensor<LayerNormOutDataType>& out_m_n,
AccDataType
acc
=
AccDataType
acc
=
static_cast
<
AccDataType
>
(
c_m_n
(
m
,
n
))
+
static_cast
<
AccDataType
>
(
bias_n
(
n
));
static_cast
<
AccDataType
>
(
c_m_n
(
m
,
n
))
+
static_cast
<
AccDataType
>
(
bias_n
(
n
));
AccDataType
c1
=
c1_m_n
(
m
,
n
);
AccDataType
c1
=
static_cast
<
AccDataType
>
(
c1_m_n
(
m
,
n
)
)
;
c_element_op
(
acc
,
acc
);
c_element_op
(
acc
,
acc
);
c1_element_op
(
c1
,
c1
);
c1_element_op
(
c1
,
c1
);
acc
+=
static_cast
<
AccDataType
>
(
c1
)
;
acc
+=
c1
;
c_m_n
(
m
,
n
)
=
static_cast
<
CDataType
>
(
acc
);
c_m_n
(
m
,
n
)
=
static_cast
<
CDataType
>
(
acc
);
}
}
...
...
profiler/include/profile_gemm_bias_add_reduce_impl.hpp
View file @
94d5f723
...
@@ -188,10 +188,10 @@ void profile_gemm_bias_add_reduce_impl(int do_verification,
...
@@ -188,10 +188,10 @@ void profile_gemm_bias_add_reduce_impl(int do_verification,
ReduceAccDataType
acc
=
static_cast
<
ReduceAccDataType
>
(
c_m_n_host_result
(
m
,
n
))
+
ReduceAccDataType
acc
=
static_cast
<
ReduceAccDataType
>
(
c_m_n_host_result
(
m
,
n
))
+
static_cast
<
ReduceAccDataType
>
(
bias_n
(
n
));
static_cast
<
ReduceAccDataType
>
(
bias_n
(
n
));
ReduceAccDataType
c1
=
c1_m_n
(
m
,
n
);
ReduceAccDataType
c1
=
static_cast
<
ReduceAccDataType
>
(
c1_m_n
(
m
,
n
)
)
;
c_element_op
(
acc
,
acc
);
c_element_op
(
acc
,
acc
);
c1_element_op
(
c1
,
c1
);
c1_element_op
(
c1
,
c1
);
acc
+=
static_cast
<
ReduceAccDataType
>
(
c1
)
;
acc
+=
c1
;
c_m_n_host_result
(
m
,
n
)
=
static_cast
<
CDataType
>
(
acc
);
c_m_n_host_result
(
m
,
n
)
=
static_cast
<
CDataType
>
(
acc
);
}
}
...
...
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