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
a4e34d88
Unverified
Commit
a4e34d88
authored
Dec 02, 2022
by
rocking5566
Committed by
GitHub
Dec 02, 2022
Browse files
Merge branch 'develop' into gemm_layernorm_welford
parents
6916e3e4
ad541ad6
Changes
63
Expand all
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
97 additions
and
8 deletions
+97
-8
test/batchnorm/batchnorm_bwd_rank_4.cpp
test/batchnorm/batchnorm_bwd_rank_4.cpp
+92
-0
test/batchnorm/batchnorm_fwd_rank_4.cpp
test/batchnorm/batchnorm_fwd_rank_4.cpp
+0
-1
test/gemm_split_k/gemm_split_k.cpp
test/gemm_split_k/gemm_split_k.cpp
+5
-7
No files found.
test/batchnorm/batchnorm_bwd_rank_4.cpp
0 → 100644
View file @
a4e34d88
This diff is collapsed.
Click to expand it.
test/batchnorm
_fwd
/batchnorm_fwd_rank_4.cpp
→
test/batchnorm/batchnorm_fwd_rank_4.cpp
View file @
a4e34d88
...
...
@@ -90,7 +90,6 @@ class TestBatchNormFwdRank4 : public ::testing::Test
using
KernelTypes
=
::
testing
::
Types
<
std
::
tuple
<
F16
,
F16
,
F32
,
F16
,
F16
,
F32
>
,
std
::
tuple
<
F32
,
F32
,
F32
,
F32
,
F32
,
F32
>
,
std
::
tuple
<
BF16
,
BF16
,
F32
,
BF16
,
BF16
,
F32
>
,
std
::
tuple
<
I8
,
I8
,
F32
,
I8
,
I8
,
F32
>
,
std
::
tuple
<
F64
,
F64
,
F64
,
F64
,
F64
,
F64
>>
;
TYPED_TEST_SUITE
(
TestBatchNormFwdRank4
,
KernelTypes
);
...
...
test/gemm_split_k/gemm_split_k.cpp
View file @
a4e34d88
This diff is collapsed.
Click to expand it.
Prev
1
2
3
4
Next
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