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
efda3cb4
Commit
efda3cb4
authored
Apr 05, 2021
by
Chao Liu
Browse files
cast void* to Type __attribute__ ((address_space(4)))*
parent
993295b9
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
11 additions
and
0 deletions
+11
-0
composable_kernel/include/tensor_operation/gridwise_dynamic_gemm.hpp
...kernel/include/tensor_operation/gridwise_dynamic_gemm.hpp
+11
-0
No files found.
composable_kernel/include/tensor_operation/gridwise_dynamic_gemm.hpp
View file @
efda3cb4
...
@@ -489,10 +489,21 @@ struct GridwiseDynamicGemm_km_kn_m0m1n0n1_v1
...
@@ -489,10 +489,21 @@ struct GridwiseDynamicGemm_km_kn_m0m1n0n1_v1
integral_constant
<
bool
,
HasMainKBlockLoop
>
,
integral_constant
<
bool
,
HasMainKBlockLoop
>
,
integral_constant
<
bool
,
HasDoubleTailKBlockLoop
>
)
const
integral_constant
<
bool
,
HasDoubleTailKBlockLoop
>
)
const
{
{
#if 0
const auto a_k_m_global_desc = *reinterpret_cast<const AGlobalDesc*>(p_a_k_m_global_desc);
const auto a_k_m_global_desc = *reinterpret_cast<const AGlobalDesc*>(p_a_k_m_global_desc);
const auto b_k_n_global_desc = *reinterpret_cast<const BGlobalDesc*>(p_b_k_n_global_desc);
const auto b_k_n_global_desc = *reinterpret_cast<const BGlobalDesc*>(p_b_k_n_global_desc);
const auto c_m0_m1_n0_n1_global_desc =
const auto c_m0_m1_n0_n1_global_desc =
*reinterpret_cast<const CGlobalDesc*>(p_c_m0_m1_n0_n1_global_desc);
*reinterpret_cast<const CGlobalDesc*>(p_c_m0_m1_n0_n1_global_desc);
#else
// cast to address_space(4)
auto
*
p_a_k_m_global_desc_const
=
(
AGlobalDesc
__attribute__
((
address_space
(
4
)))
*
)
p_a_k_m_global_desc
;
auto
*
p_b_k_n_global_desc_const
=
(
BGlobalDesc
__attribute__
((
address_space
(
4
)))
*
)
p_b_k_n_global_desc
;
auto
*
p_c_m0_m1_n0_n1_global_desc_const
=
(
CGlobalDesc
__attribute__
((
address_space
(
4
)))
*
)
p_c_m0_m1_n0_n1_global_desc
;
auto
a_k_m_global_desc
=
*
(
AGlobalDesc
*
)
p_a_k_m_global_desc_const
;
auto
b_k_n_global_desc
=
*
(
BGlobalDesc
*
)
p_b_k_n_global_desc_const
;
auto
c_m0_m1_n0_n1_global_desc
=
*
(
CGlobalDesc
*
)
p_c_m0_m1_n0_n1_global_desc_const
;
#endif
Run
(
a_k_m_global_desc
,
Run
(
a_k_m_global_desc
,
p_a_global
,
p_a_global
,
...
...
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