|
28 | 28 | EXAMPLES = [ |
29 | 29 | { |
30 | 30 | "category": "Matrix", |
31 | | - "name": "Matrix Multiplication", |
32 | | - "path": "matmul", |
| 31 | + "name": "Matrix Multiplication (BF16)", |
| 32 | + "path": "matmul_bf16_m64_n64_k64", |
33 | 33 | "datatypes": "bf16", |
34 | 34 | }, |
35 | 35 | { |
36 | 36 | "category": "Matrix", |
37 | | - "name": "Padded Matrix Multiplication", |
38 | | - "path": "padded_matmul", |
| 37 | + "name": "Padded Matrix Multiplication (F32, A Transposed)", |
| 38 | + "path": "matmul_f32_m64_n32_k16_padded_atransposed", |
39 | 39 | "datatypes": "f32 (bf16 emulation)", |
40 | 40 | }, |
| 41 | + { |
| 42 | + "category": "Matrix", |
| 43 | + "name": "Matrix Multiplication (INT8)", |
| 44 | + "path": "matmul_i8_m64_n64_k64", |
| 45 | + "datatypes": "i8", |
| 46 | + }, |
| 47 | + { |
| 48 | + "category": "Matrix", |
| 49 | + "name": "Matrix Multiplication (INT8, Large Tile)", |
| 50 | + "path": "matmul_i8_m128_n64_k64", |
| 51 | + "datatypes": "i8", |
| 52 | + }, |
41 | 53 | { |
42 | 54 | "category": "Matrix", |
43 | 55 | "name": "Matrix Multiplication (Autotune)", |
@@ -214,11 +226,11 @@ def generate_readme(base_url=""): |
214 | 226 | source /opt/xilinx/xrt/setup.sh |
215 | 227 |
|
216 | 228 | # Run an example on AIE2 (NPU1): |
217 | | -cd matmul |
218 | | -AIR_TRANSFORM_TILING_SCRIPT=transform_aie2.mlir python matmul.py |
| 229 | +cd matmul_bf16_m64_n64_k64 |
| 230 | +AIR_TRANSFORM_TILING_SCRIPT=transform_aie2.mlir python matmul_bf16_m64_n64_k64.py |
219 | 231 |
|
220 | 232 | # Run on AIE2P (NPU2): |
221 | | -AIR_TRANSFORM_TILING_SCRIPT=transform_aie2p.mlir python matmul.py |
| 233 | +AIR_TRANSFORM_TILING_SCRIPT=transform_aie2p.mlir python matmul_bf16_m64_n64_k64.py |
222 | 234 | ``` |
223 | 235 |
|
224 | 236 | ## Running All Tests |
|
0 commit comments