Skip to content

Commit b836d01

Browse files
authored
Updated Makefile for trace (#2035)
1 parent 4c71d38 commit b836d01

File tree

3 files changed

+42
-5
lines changed

3 files changed

+42
-5
lines changed

programming_examples/ml/eltwise_add/Makefile

Lines changed: 22 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,8 @@ trace_size = 8192
2020
aie_py_src=${targetname}.py
2121
use_alt?=0
2222

23+
CHESS ?= false
24+
2325
ifeq (${use_alt}, 1)
2426
aie_py_src=${targetname}_alt.py
2527
endif
@@ -28,7 +30,11 @@ VPATH := ${srcdir}/../../../aie_kernels/aie2
2830

2931
build/%.o: %.cc
3032
mkdir -p ${@D}
33+
ifeq ($(CHESS), true)
34+
cd ${@D} && ${PEANO_INSTALL_DIR}/bin/clang++ ${PEANOWRAP2_FLAGS} -c $< -o ${@F}
35+
else
3136
cd ${@D} && ${PEANO_INSTALL_DIR}/bin/clang++ ${PEANOWRAP2_FLAGS} -c $< -o ${@F}
37+
endif
3238

3339
build/aie.mlir: ${srcdir}/${aie_py_src}
3440
mkdir -p ${@D}
@@ -40,15 +46,25 @@ build/aie_trace.mlir: ${srcdir}/${aie_py_src}
4046

4147
build/final.xclbin: build/aie.mlir build/add.o
4248
mkdir -p ${@D}
49+
ifeq ($(CHESS), true)
50+
cd ${@D} && aiecc.py --aie-generate-cdo --aie-generate-npu --no-compile-host \
51+
--xclbin-name=${@F} --npu-insts-name=insts.txt ${<F}
52+
else
4353
cd ${@D} && aiecc.py --aie-generate-cdo --aie-generate-npu --no-compile-host \
4454
--no-xchesscc --no-xbridge \
4555
--xclbin-name=${@F} --npu-insts-name=insts.txt ${<F}
56+
endif
4657

4758
build/final_trace.xclbin: build/aie_trace.mlir build/add.o
4859
mkdir -p ${@D}
60+
ifeq ($(CHESS), true)
4961
cd ${@D} && aiecc.py --aie-generate-cdo --aie-generate-npu --no-compile-host \
5062
--no-xchesscc --no-xbridge \
5163
--xclbin-name=${@F} --npu-insts-name=insts.txt ${<F}
64+
else
65+
cd ${@D} && aiecc.py --aie-generate-cdo --aie-generate-npu --no-compile-host \
66+
--xclbin-name=${@F} --npu-insts-name=insts.txt ${<F}
67+
endif
5268

5369
${targetname}.exe: ${srcdir}/test.cpp
5470
rm -rf _build
@@ -67,7 +83,11 @@ run: ${targetname}.exe build/final.xclbin
6783

6884
trace: ${targetname}.exe build/final_trace.xclbin
6985
${powershell} ./$< -x build/final_trace.xclbin -i build/insts.txt -k MLIR_AIE -t ${trace_size}
70-
../../utils/parse_eventIR.py --filename trace.txt --mlir build/aie_trace.mlir --colshift 1 > parse_eventIR_vs.json
86+
../../utils/parse_trace.py --filename trace.txt --mlir build/aie_trace.mlir --colshift 1 > trace_eltwise_add.json
87+
88+
89+
clean_trace:
90+
rm -rf trace.txt trace_eltwise_add.json
7191

72-
clean:
92+
clean: clean_trace
7393
rm -rf build _build ${targetname}.exe

programming_examples/ml/relu/Makefile

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,8 @@ trace_size = 8192
2020
aie_py_src=relu.py
2121
use_alt?=0
2222

23+
CHESS ?= false
24+
2325
ifeq (${use_alt}, 1)
2426
aie_py_src=relu_alt.py
2527
endif
@@ -28,7 +30,11 @@ VPATH := ${srcdir}/../../../aie_kernels/aie2
2830

2931
build/%.o: %.cc
3032
mkdir -p ${@D}
33+
ifeq ($(CHESS), true)
34+
cd ${@D} && xchesscc_wrapper ${CHESSCCWRAP2_FLAGS} -c $< -o ${@F};
35+
else
3136
cd ${@D} && ${PEANO_INSTALL_DIR}/bin/clang++ ${PEANOWRAP2_FLAGS} -c $< -o ${@F}
37+
endif
3238

3339
build/aie.mlir: ${srcdir}/${aie_py_src}
3440
mkdir -p ${@D}
@@ -46,9 +52,14 @@ build/final.xclbin: build/aie.mlir build/relu.o
4652

4753
build/final_trace.xclbin: build/aie_trace.mlir build/relu.o
4854
mkdir -p ${@D}
55+
ifeq ($(CHESS), true)
56+
cd ${@D} && aiecc.py --aie-generate-cdo --aie-generate-npu --no-compile-host \
57+
--xclbin-name=${@F} --npu-insts-name=insts.txt ${<F}
58+
else
4959
cd ${@D} && aiecc.py --aie-generate-cdo --aie-generate-npu --no-compile-host \
5060
--no-xchesscc --no-xbridge \
5161
--xclbin-name=${@F} --npu-insts-name=insts.txt ${<F}
62+
endif
5263

5364
${targetname}.exe: ${srcdir}/test.cpp
5465
rm -rf _build
@@ -68,5 +79,8 @@ trace: ${targetname}.exe build/final_trace.xclbin
6879
${powershell} ./$< -x build/final_trace.xclbin -i build/insts.txt -k MLIR_AIE -t ${trace_size}
6980
../../utils/parse_trace.py --filename trace.txt --mlir build/aie_trace.mlir --colshift 1 > trace_relu.json
7081

71-
clean:
82+
clean_trace:
83+
rm -rf trace.txt trace_relu.json
84+
85+
clean: clean_trace
7286
rm -rf build _build ${targetname}.exe

programming_examples/ml/softmax/Makefile

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,8 +85,11 @@ profile: ${targetname}.exe build/final.xclbin
8585

8686
trace: ${targetname}.exe build/final_trace.xclbin
8787
${powershell} ./$< -x build/final_trace.xclbin -i build/insts.txt -k MLIR_AIE -t ${trace_size}
88-
../../utils/parse_eventIR.py --filename trace.txt --mlir build/aie_trace.mlir --colshift 1 > parse_eventIR_vs.json
88+
../../utils/parse_trace.py --filename trace.txt --mlir build/aie_trace.mlir --colshift 1 > trace_softmax.json
8989

90-
clean:
90+
clean_trace:
91+
rm -rf trace.txt trace_softmax.json
92+
93+
clean: clean_trace
9194
rm -rf build _build ${targetname}.exe
9295

0 commit comments

Comments
 (0)