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
6bc73d41
Commit
6bc73d41
authored
Sep 25, 2023
by
letaoqin
Browse files
rename example file name
parent
d9d68abf
Changes
6
Expand all
Hide whitespace changes
Inline
Side-by-side
Showing
6 changed files
with
498 additions
and
198 deletions
+498
-198
example/52_flash_atten_bias/batched_gemm_multihead_attention_bias_forward.cpp
...en_bias/batched_gemm_multihead_attention_bias_forward.cpp
+1
-1
example/52_flash_atten_bias/batched_gemm_multihead_attention_forward.cpp
...h_atten_bias/batched_gemm_multihead_attention_forward.cpp
+1
-1
example/52_flash_atten_bias/batched_multihead_attention_bias_forward_v2.cpp
...tten_bias/batched_multihead_attention_bias_forward_v2.cpp
+1
-1
example/52_flash_atten_bias/run_batched_multihead_attention_bias_forward.inc
...ten_bias/run_batched_multihead_attention_bias_forward.inc
+66
-195
example/52_flash_atten_bias/run_batched_multihead_attention_bias_forward_v2.inc
..._bias/run_batched_multihead_attention_bias_forward_v2.inc
+429
-0
example/52_flash_atten_bias/run_batched_multihead_attention_forward.inc
...sh_atten_bias/run_batched_multihead_attention_forward.inc
+0
-0
No files found.
example/52_flash_atten_bias/batched_gemm_multihead_attention_bias_forward.cpp
View file @
6bc73d41
...
@@ -156,6 +156,6 @@ using ReferenceGemm1Instance = ck::tensor_operation::host::ReferenceBatchedGemm<
...
@@ -156,6 +156,6 @@ using ReferenceGemm1Instance = ck::tensor_operation::host::ReferenceBatchedGemm<
B1ElementOp
,
B1ElementOp
,
CElementOp
>
;
CElementOp
>
;
#include "run_batched_mutihead_attention_bias_forward.inc"
#include "run_batched_mu
l
tihead_attention_bias_forward.inc"
int
main
(
int
argc
,
char
*
argv
[])
{
return
run
(
argc
,
argv
);
}
int
main
(
int
argc
,
char
*
argv
[])
{
return
run
(
argc
,
argv
);
}
example/52_flash_atten_bias/batched_gemm_multihead_attention_forward.cpp
View file @
6bc73d41
...
@@ -156,6 +156,6 @@ using ReferenceGemm1Instance = ck::tensor_operation::host::ReferenceBatchedGemm<
...
@@ -156,6 +156,6 @@ using ReferenceGemm1Instance = ck::tensor_operation::host::ReferenceBatchedGemm<
B1ElementOp
,
B1ElementOp
,
CElementOp
>
;
CElementOp
>
;
#include "run_batched_mutihead_attention_forward.inc"
#include "run_batched_mu
l
tihead_attention_forward.inc"
int
main
(
int
argc
,
char
*
argv
[])
{
return
run
(
argc
,
argv
);
}
int
main
(
int
argc
,
char
*
argv
[])
{
return
run
(
argc
,
argv
);
}
example/52_flash_atten_bias/batched_multihead_attention_bias_forward_v2.cpp
View file @
6bc73d41
...
@@ -327,6 +327,6 @@ using ReferenceGemm1Instance = ck::tensor_operation::host::ReferenceBatchedGemm<
...
@@ -327,6 +327,6 @@ using ReferenceGemm1Instance = ck::tensor_operation::host::ReferenceBatchedGemm<
using
ReferenceDropoutInstance
=
using
ReferenceDropoutInstance
=
ck
::
tensor_operation
::
host
::
ReferenceDropout
<
ZDataType
,
ADataType
,
ADataType
>
;
ck
::
tensor_operation
::
host
::
ReferenceDropout
<
ZDataType
,
ADataType
,
ADataType
>
;
#include "run_batched_multihead_attention_bias_forward.inc"
#include "run_batched_multihead_attention_bias_forward
_v2
.inc"
int
main
(
int
argc
,
char
*
argv
[])
{
return
run
(
argc
,
argv
);
}
int
main
(
int
argc
,
char
*
argv
[])
{
return
run
(
argc
,
argv
);
}
example/52_flash_atten_bias/run_batched_multihead_attention_bias_forward.inc
View file @
6bc73d41
This diff is collapsed.
Click to expand it.
example/52_flash_atten_bias/run_batched_mutihead_attention_bias_forward.inc
→
example/52_flash_atten_bias/run_batched_mu
l
tihead_attention_bias_forward
_v2
.inc
View file @
6bc73d41
This diff is collapsed.
Click to expand it.
example/52_flash_atten_bias/run_batched_mutihead_attention_forward.inc
→
example/52_flash_atten_bias/run_batched_mu
l
tihead_attention_forward.inc
View file @
6bc73d41
File moved
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