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
yangql
composable_kernel-1
Commits
2cbb8976
"...git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "90f7a7d9e5d785333f412601e12097563d2260fc"
Commit
2cbb8976
authored
Dec 02, 2021
by
Jing Zhang
Browse files
add static_buffer_v2 zero out
parent
d798c9b8
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
15 additions
and
0 deletions
+15
-0
composable_kernel/include/tensor_operation/gridwise_gemm_xdlops_v2r3.hpp
...el/include/tensor_operation/gridwise_gemm_xdlops_v2r3.hpp
+2
-0
composable_kernel/include/utility/static_buffer_of_vector_type_v2.hpp
...ernel/include/utility/static_buffer_of_vector_type_v2.hpp
+13
-0
No files found.
composable_kernel/include/tensor_operation/gridwise_gemm_xdlops_v2r3.hpp
View file @
2cbb8976
...
...
@@ -518,6 +518,8 @@ struct GridwiseGemm_k0mk1_k0nk1_mn_xdlops_v2r3
// main body
index_t
k0_block_data_begin
=
0
;
c_thread_buf
.
Clear
();
if
constexpr
(
HasMainKBlockLoop
)
{
do
...
...
composable_kernel/include/utility/static_buffer_of_vector_type_v2.hpp
View file @
2cbb8976
...
...
@@ -22,6 +22,13 @@ struct StaticBufferOfVectorTypeV2 : public StaticallyIndexedArray<T, N>
static
constexpr
index_t
vector_size
=
GetVectorSize
();
__host__
__device__
static
constexpr
index_t
GetNumVectors
()
{
return
N
;
}
__host__
__device__
static
constexpr
index_t
GetNumElements
()
{
return
GetVectorSize
()
*
GetNumVectors
();
}
VecBaseType
invalid_element_value_
=
VecBaseType
{
0
};
T
invalid_vec_value_
=
T
{
0
};
...
...
@@ -91,6 +98,12 @@ struct StaticBufferOfVectorTypeV2 : public StaticallyIndexedArray<T, N>
return
GetElement
(
i
,
true
);
}
__host__
__device__
void
Clear
()
{
static_for
<
0
,
GetNumElements
(),
1
>
{}(
[
&
](
auto
i
)
{
GetElement
(
i
,
true
)
=
invalid_element_value_
;
});
}
__host__
__device__
static
constexpr
bool
IsStaticBuffer
()
{
return
true
;
}
__host__
__device__
static
constexpr
bool
IsDynamicBuffer
()
{
return
false
;
}
...
...
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