Commit cca078a
authored
File tree
- .github/workflows
- client_example
- 01_gemm
- 02_gemm_add_add_fastgelu
- 03_gemm_layernorm
- 04_contraction
- 05_layernorm
- 06_softmax
- 07_grouped_convnd_fwd
- 08_fused_attention
- 09_quantization
- 10_grouped_convnd_bwd_data
- 11_grouped_conv_bwd_weight
- 12_elementwise_normalization
- 13_batchnorm
- 14_instance_id
- 15_convnd_bwd_data
- 16_convnd_fwd
- 17_grouped_gemm_fastgelu
- 18_groupnorm
- 19_pool
- 20_splitk_gemm
- 21_grouped_gemm_bias
- 22_grouped_gemm
- 23_elementwise_transpose
- 24_grouped_conv_activation
- grouped_convnd_bwd_data_bilinear
- grouped_convnd_bwd_data_scale
- grouped_convnd_bwd_weight_bilinear
- grouped_convnd_bwd_weight_scale
- grouped_convnd_fwd_bilinear
- grouped_convnd_fwd_convinvscale
- grouped_convnd_fwd_convscale_add
- grouped_convnd_fwd_convscale_reduce
- grouped_convnd_fwd_convscale_relu
- grouped_convnd_fwd_convscale
- grouped_convnd_fwd_scaleadd_ab
- grouped_convnd_fwd_scaleadd_scaleadd_relu
- grouped_convnd_fwd_scale
- 25_wrapper
- 26_reduce
- 27_im2col_col2im
- 28_gemm_mx
- 29_gemm_add_multiply
- 30_gemm_bf16Aint8B
- 31_grouped_gemm_bf16Aint8B
- cmake
- codegen
- driver
- include/ck/host
- device_batched_gemm_softmax_gemm
- device_gemm_multiple_d
- device_grouped_conv_fwd_multiple_d
- operation
- src
- test
- include
- rtc
- include/rtc
- src
- docs
- example
- 01_gemm
- 02_gemm_bilinear
- 03_gemm_bias_relu
- 04_gemm_add_add_fastgelu
- 09_convnd_fwd
- 10_convnd_fwd_multiple_d_multiple_reduce
- 12_reduce
- 13_pool2d_fwd
- 14_gemm_quantization
- 15_grouped_gemm
- 16_gemm_multi_d_multi_reduces
- 17_convnd_bwd_data
- 18_batched_gemm_reduce
- 19_binary_elementwise
- 20_grouped_conv_bwd_weight
- 21_gemm_layernorm
- 22_cgemm
- 23_softmax
- 24_batched_gemm
- 25_gemm_bias_e_permute
- 26_contraction
- 27_layernorm2d_fwd
- 28_grouped_gemm_bias_e_permute
- 29_batched_gemm_bias_e_permute
- 30_grouped_conv_fwd_multiple_d
- 31_batched_gemm_gemm
- 32_batched_gemm_scale_softmax_gemm
- 33_multiple_reduce
- 34_batchnorm
- 35_splitK_gemm
- 36_sparse_embedding
- 37_batched_gemm_add_add_relu_gemm_add
- 38_grouped_conv_bwd_data_multiple_d
- 39_permute
- 40_conv2d_fwd_quantization
- 41_grouped_conv_conv_fwd
- 42_groupnorm_fwd
- 43_splitk_gemm_bias_e_permute
- 44_elementwise_permute
- 45_elementwise_normalization
- 46_gemm_add_multiply
- 47_gemm_bias_softmax_gemm_permute
- 48_pool3d_fwd
- 49_maxpool2d_bwd
- 50_put_element
- 51_avgpool3d_bwd
- 52_im2col_col2im
- 53_layernorm2d_bwd
- 54_groupnorm_bwd
- 59_grouped_gemm_multi_ABD
- 60_gemm_multi_ABD
- 61_contraction_multi_ABD
- 62_convnd_activ
- binary
- convinvscale
- convscale_add
- convscale_reduce
- convscale_relu
- convscale
- dynamic_unary
- multi_AB
- unary
- 63_layernorm4d_fwd
- 64_fpAintB_gemm
- 65_gemm_multiply_multiply
- 66_complex_contraction_bilinear
- 67_gemm_microscaling
- 68_gemm_add
- 69_gemm_add_relu
- ck_tile
- 01_fmha
- codegen
- ops
- instances
- script
- 02_layernorm2d
- script
- 03_gemm
- script
- 04_img2col
- 05_reduce
- 06_permute
- alternative_impl
- script
- 09_topk_softmax
- script
- 10_rmsnorm2d
- script
- 11_add_rmsnorm2d_rdquant
- instances
- script
- 12_smoothquant
- instances
- script
- 13_moe_sorting
- script
- 14_moe_smoothquant
- instances
- script
- 15_fused_moe
- instances
- 16_batched_gemm
- 17_grouped_gemm
- 18_flatmm
- mixed_prec
- mxgemm
- script
- 19_gemm_multi_d
- 20_grouped_convolution
- 21_elementwise
- 22_gemm_multi_abd
- 35_batched_transpose
- script
- 36_pooling
- 38_block_scale_gemm
- 39_copy
- 40_streamk_gemm
- 41_batched_contraction
- experimental/builder
- include/ck_tile/builder
- reflect
- test
- conv
- impl
- utils
- include
- ck_tile
- core
- algorithm
- arch
- mma
- mfma
- wmma
- container
- numeric
- tensor
- utility
- host
- reference
- ops
- add_rmsnorm2d_rdquant
- kernel
- pipeline
- batched_contraction
- kernel
- pipeline
- utils
- batched_transpose
- kernel
- pipeline
- common
- elementwise
- kernel
- pipeline
- epilogue
- flatmm
- block
- uk
- kernel
- pipeline
- fmha
- block
- kernel
- pipeline
- fused_moe
- kernel
- pipeline
- gemm_quant
- block
- kernel
- pipeline
- gemm
- block
- kernel
- pipeline
- warp
- grouped_convolution
- kernel
- utils
- image_to_column
- kernel
- pipeline
- layernorm2d
- kernel
- pipeline
- norm_reduce
- block
- thread
- permute
- kernel
- pipeline
- pooling
- kernel
- pipeline
- reduce
- block
- kernel
- pipeline
- rmsnorm2d
- kernel
- pipeline
- smoothquant
- kernel
- pipeline
- softmax/block
- topk_softmax
- kernel
- pipeline
- topk/block
- ref
- utility
- ck
- host_utility
- library/utility
- problem_transform
- tensor_description
- tensor_operation
- gpu
- block
- device
- impl
- element
- grid
- batchnorm_multiblock
- gemm_layernorm
- normalization
- thread
- warp
- operator_transform
- tensor
- utility
- wrapper
- operations
- traits
- utils
- library
- include/ck/library
- reference_tensor_operation
- cpu
- gpu
- tensor_operation_instance
- gpu
- contraction
- conv_tensor_rearrange
- grouped_conv_bwd_data
- grouped_conv_bwd_weight
- grouped_conv_fwd
- grouped_gemm
- permute_scale
- quantization
- reduce
- softmax
- transpose
- src/tensor_operation_instance/gpu
- avg_pool2d_bwd
- avg_pool3d_bwd
- batched_gemm_add_relu_gemm_add
- batched_gemm_b_scale
- device_batched_gemm_b_scale_wmma_f16_i4_f16
- device_batched_gemm_b_scale_xdl_f16_i4_f16
- batched_gemm_bias_permute
- batched_gemm_gemm
- batched_gemm_multi_d
- batched_gemm_reduce
- batched_gemm_softmax_gemm_permute
- batched_gemm_softmax_gemm
- batched_gemm
- batchnorm
- column_to_image
- contraction_bilinear
- 2D
- 6D
- contraction_scale
- 2D
- 6D
- conv1d_bwd_data
- conv2d_bwd_data
- conv2d_fwd_bias_relu_add
- conv2d_fwd_bias_relu
- conv2d_fwd
- conv3d_bwd_data
- elementwise_normalization
- elementwise
- gemm_ab_scale
- device_gemm_ab_scale_xdl_f8_f8_bf16
- gemm_add_add_fastgelu
- gemm_add_fastgelu
- gemm_add_multiply
- gemm_add_relu_add_layernorm
- gemm_add_relu
- gemm_add_silu
- gemm_add
- gemm_b_scale
- device_gemm_b_scale_wmma_f16_i4_f16
- device_gemm_b_scale_xdl_f16_i4_f16
- gemm_bias_add_reduce
- gemm_bilinear
- gemm_blockscale_wp
- device_gemm_blockscale_wp_xdl_f8_f8_bf16
- gemm_fastgelu
- gemm_multi_abd
- gemm_multiply_add
- gemm_multiply_multiply_wp
- f8_f8_bf16
- f8_f8_f16
- gemm_multiply_multiply
- device_gemm_multiply_multiply_xdl_f8_f8_bf16
- device_gemm_multiply_multiply_xdl_f8_f8_f16
- device_gemm_multiply_multiply_xdl_i8_i8_bf16
- device_gemm_multiply_multiply_xdl_i8_i8_f16
- gemm_mx
- device_gemm_mx_xdl_bf6_bf6_bf16
- device_gemm_mx_xdl_bf8_f8_f16
- device_gemm_mx_xdl_f4_f4_f16
- device_gemm_mx_xdl_f6_f6_f16
- device_gemm_mx_xdl_f8_f8_bf16
- device_gemm_mx_xdl_f8_f8_f16
- gemm
- device_gemm_xdl_f16_f16_f16
Some content is hidden
Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
53 | 53 | | |
54 | 54 | | |
55 | 55 | | |
56 | | - | |
57 | 56 | | |
| 57 | + | |
58 | 58 | | |
59 | 59 | | |
60 | 60 | | |
| |||
77 | 77 | | |
78 | 78 | | |
79 | 79 | | |
| 80 | + | |
| 81 | + | |
80 | 82 | | |
81 | 83 | | |
82 | 84 | | |
| |||
128 | 130 | | |
129 | 131 | | |
130 | 132 | | |
131 | | - | |
| 133 | + | |
132 | 134 | | |
133 | 135 | | |
134 | 136 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
51 | 51 | | |
52 | 52 | | |
53 | 53 | | |
54 | | - | |
| 54 | + | |
55 | 55 | | |
56 | 56 | | |
57 | 57 | | |
58 | 58 | | |
59 | 59 | | |
60 | | - | |
| 60 | + | |
61 | 61 | | |
62 | 62 | | |
63 | 63 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
27 | 27 | | |
28 | 28 | | |
29 | 29 | | |
30 | | - | |
| 30 | + | |
31 | 31 | | |
32 | 32 | | |
33 | 33 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
2 | 2 | | |
3 | 3 | | |
4 | 4 | | |
5 | | - | |
6 | | - | |
7 | | - | |
| 5 | + | |
8 | 6 | | |
9 | 7 | | |
| 8 | + | |
10 | 9 | | |
11 | 10 | | |
12 | 11 | | |
| |||
32 | 31 | | |
33 | 32 | | |
34 | 33 | | |
| 34 | + | |
| 35 | + | |
| 36 | + | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
| 41 | + | |
| 42 | + | |
| 43 | + | |
35 | 44 | | |
36 | 45 | | |
37 | 46 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
1 | 4 | | |
2 | 5 | | |
3 | 6 | | |
| |||
39 | 42 | | |
40 | 43 | | |
41 | 44 | | |
| 45 | + | |
| 46 | + | |
42 | 47 | | |
43 | 48 | | |
44 | 49 | | |
| |||
122 | 127 | | |
123 | 128 | | |
124 | 129 | | |
125 | | - | |
| 130 | + | |
126 | 131 | | |
127 | 132 | | |
128 | 133 | | |
| |||
229 | 234 | | |
230 | 235 | | |
231 | 236 | | |
232 | | - | |
| 237 | + | |
233 | 238 | | |
234 | 239 | | |
235 | 240 | | |
236 | 241 | | |
237 | | - | |
| 242 | + | |
238 | 243 | | |
239 | 244 | | |
240 | 245 | | |
| |||
247 | 252 | | |
248 | 253 | | |
249 | 254 | | |
250 | | - | |
| 255 | + | |
251 | 256 | | |
252 | 257 | | |
253 | 258 | | |
| |||
257 | 262 | | |
258 | 263 | | |
259 | 264 | | |
260 | | - | |
| 265 | + | |
261 | 266 | | |
262 | 267 | | |
263 | 268 | | |
| |||
683 | 688 | | |
684 | 689 | | |
685 | 690 | | |
| 691 | + | |
| 692 | + | |
| 693 | + | |
| 694 | + | |
| 695 | + | |
| 696 | + | |
686 | 697 | | |
687 | 698 | | |
688 | 699 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
17 | 17 | | |
18 | 18 | | |
19 | 19 | | |
| 20 | + | |
| 21 | + | |
20 | 22 | | |
0 commit comments