ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 18:17:44 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
f913b45773bba61ae902deaa170a3012c44cf099
composable_kernel/example
History
ltqin 8c5f5f1293 Fix flash attn mask bug (#733) 
* add check input parameter

* add instance for vector load = 1

* move gerneral instance to first pos

* fix read bias code

* regular code for bias load

---------

Co-authored-by: zjing14 <zhangjing14@gmail.com>

[ROCm/composable_kernel commit: 0ede66de54]
2023-06-12 08:35:31 -05:00
..
01_gemm
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
02_gemm_bilinear
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
03_gemm_bias_relu
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
04_gemm_add_add_fastgelu
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
09_convnd_fwd
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
10_convnd_fwd_multiple_d_multiple_reduce
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
12_reduce
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
13_pool2d_fwd
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
14_gemm_quantization
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
15_grouped_gemm
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
16_gemm_multi_d_multi_reduces
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
17_convnd_bwd_data
Simplify kernel argument of device operator Device(Batched)GemmXdl<> (#723)
2023-06-01 16:23:02 -05:00
18_batched_gemm_reduce
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
19_binary_elementwise
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
20_grouped_conv_bwd_weight
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
21_gemm_layernorm
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
22_cgemm
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
23_softmax
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
24_batched_gemm
Rangify constructor of HostTensorDescriptor & Tensor<> (#445)
2022-11-11 11:36:01 -06:00
25_gemm_bias_e_permute
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
26_contraction
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
27_layernorm
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
28_grouped_gemm_bias_e_permute
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
29_batched_gemm_bias_e_permute
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
30_grouped_conv_fwd_multiple_d
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
31_batched_gemm_gemm
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
32_batched_gemm_scale_softmax_gemm
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
33_multiple_reduce
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
34_batchnorm
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
35_splitK_gemm
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
36_sparse_embedding
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
37_batched_gemm_add_add_relu_gemm_add
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
38_grouped_conv_bwd_data_multiple_d
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
39_permute
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
40_conv2d_fwd_quantization
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
41_grouped_conv_conv_fwd
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
42_groupnorm
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
43_splitk_gemm_bias_e_permute
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
44_elementwise_permute
Batchnorm inference instances, external API, client examples and gtests (#531)
2023-01-25 17:09:04 -06:00
45_elementwise_normalization
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
46_gemm_add_multiply
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
47_gemm_bias_softmax_gemm_permute
Fix flash attn mask bug (#733)
2023-06-12 08:35:31 -05:00
48_pool3d_fwd
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
CMakeLists.txt
Add packages for examples and profiler (#502)
2022-11-10 13:19:33 -06:00