This repository was archived by the owner on Aug 30, 2024. It is now read-only.
File tree
286 files changed
+49462
-42696
lines changed- examples
- 01_gemm_universal
- 02_basic_gemm
- 03_gemm_relu_bias
- 04_gemm_polynomial
- 05_batch_gemm
- 06_gemm_softmax
- 07_multi_layer_perceptron
- 08_scaled_dot_product_attention
- 09_gate_recurrent_unit
- 10_gemm_large_n
- 11_stream_k_gemm
- include
- common
- core
- utils
- experimental
- common
- group
- fused_op
- gemm
- impl
- reduction
- kernel
- data_transformer
- gemm
- impl
- layer_norm
- mha_core_attention
- reduction
- subgroup
- group
- epilogue
- impl
- gemm
- impl
- reduction
- softmax
- impl
- kernel
- default_config
- gemm
- impl
- subgroup
- tile
- impl
- tests
- integration
- data_transformer
- default_config
- group_gemm
- kernel_gemm
- gemm
- bf16_stream_k
- bf16
- fp16
- fp32
- int4_dequantization_bias
- int4_dequantization
- int8_quantization
- int8
- tf32
- unaligned_bf16
- layer_norm
- backward
- forward
- limitation
- row_reduction
- sg_dropout_op
- softmax
- vector_add
- bf16_2d
- int32_1d
- int32_2d
- tf32_1d
- unit
- add_c
- bit_mask_manipulation
- block_load_store
- buff_compare
- epilogue_tile_op
- exp_inv_sqrt_tanh
- global_atomic
- global_load_store
- imul
- internal_type_load_store_cvt
- local_load_store
- math_general
- named_barrier
- philox_rng
- raw_send
- reg_layout_conversion
- reg_reduce
- tile_load_store_local
- tile_load_store
- tile_mma
- tile_row_reduction
- utils
- tools/clang-format
Some content is hidden
Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.
286 files changed
+49462
-42696
lines changedLines changed: 88 additions & 0 deletions
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
| 10 | + | |
| 11 | + | |
| 12 | + | |
| 13 | + | |
| 14 | + | |
| 15 | + | |
| 16 | + | |
| 17 | + | |
| 18 | + | |
| 19 | + | |
| 20 | + | |
| 21 | + | |
| 22 | + | |
| 23 | + | |
| 24 | + | |
| 25 | + | |
| 26 | + | |
| 27 | + | |
| 28 | + | |
| 29 | + | |
| 30 | + | |
| 31 | + | |
| 32 | + | |
| 33 | + | |
| 34 | + | |
| 35 | + | |
| 36 | + | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
| 41 | + | |
| 42 | + | |
| 43 | + | |
| 44 | + | |
| 45 | + | |
| 46 | + | |
| 47 | + | |
| 48 | + | |
| 49 | + | |
| 50 | + | |
| 51 | + | |
| 52 | + | |
| 53 | + | |
| 54 | + | |
| 55 | + | |
| 56 | + | |
| 57 | + | |
| 58 | + | |
| 59 | + | |
| 60 | + | |
| 61 | + | |
| 62 | + | |
| 63 | + | |
| 64 | + | |
| 65 | + | |
| 66 | + | |
| 67 | + | |
| 68 | + | |
| 69 | + | |
| 70 | + | |
| 71 | + | |
| 72 | + | |
| 73 | + | |
| 74 | + | |
| 75 | + | |
| 76 | + | |
| 77 | + | |
| 78 | + | |
| 79 | + | |
| 80 | + | |
| 81 | + | |
| 82 | + | |
| 83 | + | |
| 84 | + | |
| 85 | + | |
| 86 | + | |
| 87 | + | |
| 88 | + |
0 commit comments