PTQ with ResNet conv2x layers

programming_examples/ml/resnet/ptq_conv2x/CMakeLists.txt

@@ -0,0 +1,89 @@
+# This file is licensed under the Apache License v2.0 with LLVM Exceptions.
+# See https://llvm.org/LICENSE.txt for license information.
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+#
+# Copyright (C) 2024, Advanced Micro Devices, Inc.
+
+# parameters
+# -DBOOST_ROOT: Path to Boost install
+# -DOpenCV_DIR: Path to OpenCV install
+# -DXRT_INC_DIR: Full path to src/runtime_src/core/include in XRT cloned repo
+# -DXRT_LIB_DIR: Path to xrt_coreutil.lib
+# -DTARGET_NAME: Target name to be built
+
+# cmake needs this line
+cmake_minimum_required(VERSION 3.1)
+
+find_program(WSL NAMES powershell.exe)
+
+if (NOT WSL)
+    set(BOOST_ROOT /usr/include/boost CACHE STRING "Path to Boost install")
+    set(OpenCV_DIR /usr/include/opencv4 CACHE STRING "Path to OpenCV install")
+    set(XRT_INC_DIR /opt/xilinx/xrt/include CACHE STRING "Path to XRT cloned repo")
+    set(XRT_LIB_DIR /opt/xilinx/xrt/lib CACHE STRING "Path to xrt_coreutil.lib")
+else()
+    set(BOOST_ROOT C:/Technical/thirdParty/boost_1_83_0 CACHE STRING "Path to Boost install")
+    set(OpenCV_DIR C:/Technical/thirdParty/opencv/build CACHE STRING "Path to OpenCV install")
+    set(XRT_INC_DIR C:/Technical/XRT/src/runtime_src/core/include CACHE STRING "Path to XRT cloned repo")
+    set(XRT_LIB_DIR C:/Technical/xrtNPUfromDLL CACHE STRING "Path to xrt_coreutil.lib")
+endif ()
+
+set(EDGEDETECT_WIDTH 1920 CACHE STRING "image width")
+set(EDGEDETECT_HEIGHT 1080 CACHE STRING "image height")
+
+set(TARGET_NAME test CACHE STRING "Target to be built")
+
+SET (ProjectName ${TARGET_NAME})
+SET (currentTarget ${TARGET_NAME})
+
+if ( WSL )
+	set(CMAKE_RUNTIME_OUTPUT_DIRECTORY_RELEASE ${CMAKE_BINARY_DIR})
+endif ()
+
+project(${ProjectName})
+
+# Find packages
+find_package(Boost REQUIRED)
+find_package(OpenCV REQUIRED)
+message("opencv library paht: ${OpenCV_LIB_PATH}")
+message("opencv libs: ${OpenCV_LIBS}")
+
+
+add_executable(${currentTarget}
+    ${CMAKE_CURRENT_SOURCE_DIR}/../../../utils/OpenCVUtils.cpp
+    ${CMAKE_CURRENT_SOURCE_DIR}/../../../utils/xrtUtils.cpp
+    test.cpp
+)
+
+target_compile_definitions(${currentTarget} PUBLIC 
+        EDGEDETECT_WIDTH=${EDGEDETECT_WIDTH}
+        EDGEDETECT_HEIGHT=${EDGEDETECT_HEIGHT}
+        DISABLE_ABI_CHECK=1 
+        )
+
+target_include_directories (${currentTarget} PUBLIC 
+    ${CMAKE_CURRENT_SOURCE_DIR}/../../../utils
+    ${XRT_INC_DIR}
+    ${OpenCV_INCLUDE_DIRS}
+    ${Boost_INCLUDE_DIRS}
+)
+
+target_link_directories(${currentTarget} PUBLIC
+    ${XRT_LIB_DIR}
+    ${OpenCV_LIB_PATH}
+    ${Boost_LIBRARY_DIRS}
+)
+
+if (NOT WSL)
+    target_link_libraries(${currentTarget} PUBLIC
+        xrt_coreutil
+        ${OpenCV_LIBS}
+        boost_program_options
+        boost_filesystem
+    )
+else()
+    target_link_libraries(${currentTarget} PUBLIC
+        xrt_coreutil
+        ${OpenCV_LIBS}
+    )
+endif()

programming_examples/ml/resnet/ptq_conv2x/Makefile

@@ -0,0 +1,49 @@
+##===- Makefile -----------------------------------------------------------===##
+# 
+# This file licensed under the Apache License v2.0 with LLVM Exceptions.
+# See https://llvm.org/LICENSE.txt for license information.
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+# 
+##===----------------------------------------------------------------------===##
+
+srcdir := $(shell dirname $(realpath $(firstword $(MAKEFILE_LIST))))
+
+include ${srcdir}/../../../makefile-common
+
+mlirFileName = aie
+
+VPATH := ${srcdir}/../../../../aie_kernels/aie2
+
+all: build/conv2dk1_i8.o build/conv2dk1_skip_init.o build/conv2dk3.o build/conv2dk1_skip.o build/conv2dk1_ui8.o build/final.xclbin
+
+build/${mlirFileName}.mlir: ${srcdir}/aie2.py
+	mkdir -p ${@D}
+	python3 $< > $@
+insts.txt: build/${mlirFileName}.mlir
+	aiecc.py -v --aie-only-generate-npu --npu-insts-name=$@ $<
+
+build/conv2dk1_i8.o: conv2dk1.cc
+	xchesscc -d ${CHESSCC2_FLAGS} -DINT8_ACT -c $< -o $@
+
+build/conv2dk3.o: conv2dk3.cc
+	xchesscc -d ${CHESSCC2_FLAGS} -DUINT8_ACT -c $< -o $@
+
+build/conv2dk1_skip_init.o: conv2dk1_skip_init.cc
+	xchesscc -d ${CHESSCC2_FLAGS} -DINT8_ACT -c $< -o $@
+
+build/conv2dk1_ui8.o: conv2dk1.cc
+	xchesscc -d ${CHESSCC2_FLAGS} -DUINT8_ACT -c $< -o $@
+
+build/conv2dk1_skip.o: conv2dk1_skip.cc
+	xchesscc -d ${CHESSCC2_FLAGS} -DUINT8_ACT -c $< -o $@
+
+build/final.xclbin: build/${mlirFileName}.mlir build/conv2dk1_i8.o build/conv2dk1_skip_init.o build/conv2dk3.o build/conv2dk1_skip.o build/conv2dk1_ui8.o
+	cd build && aiecc.py --basic-alloc-scheme --aie-generate-cdo --aie-generate-npu --no-compile-host \
+		--xclbin-name=${@F} --npu-insts-name=insts.txt ${<F}
+clean:
+	rm -rf  build/*.elf* build/*.lst build/*.bif log* build/${mlirFileName}.mlir.prj build/*.xclbin sim \
+		build/chess* build/insts.txt \
+		build/*.log build/aie_partition.json build/*.bin build/BOOT.BIN _x test.exe
+
+run_py: 
+	${powershell} python3 ${srcdir}/test.py -x build/final.xclbin -i build/insts.txt -k MLIR_AIE