Skip to content

Commit 176e7ab

Browse files
Mat mul whole array implementation using tiler helper tools (#1924)
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
1 parent 057ef6b commit 176e7ab

File tree

10 files changed

+689
-19
lines changed

10 files changed

+689
-19
lines changed

programming_examples/basic/matrix_multiplication/whole_array/Makefile

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,11 @@ kernels=mm_${m}x${k}x${n}
2424
aieargs+=-m $m -k $k -n $n --n-aie-cols ${n_aie_cols} --b-col-maj ${b_col_maj}
2525
runargs+=--b_col_maj ${b_col_maj}
2626
target_suffix=${M}x${K}x${N}_${m}x${k}x${n}_${n_aie_cols}c
27+
use_tiler?=0
28+
29+
ifeq (${use_tiler}, 1)
30+
aie_py_src=aie2_tiler.py
31+
endif
2732

2833
include ${srcdir}/../makefile-common
2934

programming_examples/basic/matrix_multiplication/whole_array/Makefile.chess

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,12 @@ include ${srcdir}/../makefile-common
2929

3030
CHESS=true
3131

32+
use_tiler?=0
33+
34+
ifeq (${use_tiler}, 1)
35+
aie_py_src=aie2_tiler.py
36+
endif
37+
3238
build/mm_b_row_maj_${m}x${k}x${n}.o: ${kernels_dir}/mm.cc
3339
mkdir -p ${@D}
3440
cd ${@D} && xchesscc_wrapper ${CHESSCCWRAP2_FLAGS} -DBIT_WIDTH=8 -D${dtype_in}_${dtype_out}_ONLY -DDIM_M=${m} -DDIM_K=${k} -DDIM_N=${n} -c $< -o ${@F}

0 commit comments

Comments
 (0)