From ba07af5e51e0d7d38f31890cb39669e01ce7d2aa Mon Sep 17 00:00:00 2001 From: Peng Date: Fri, 31 Aug 2018 19:33:41 -0500 Subject: [PATCH] Update Makefile to add rocm include path This commit add /opt/rocm/include to the makefile. The MIOpen and rocBLAS header files can be located under /opt/rocm/include directory. --- code/amd/Makefile | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/code/amd/Makefile b/code/amd/Makefile index 6860a55..eccf1b3 100644 --- a/code/amd/Makefile +++ b/code/amd/Makefile @@ -5,6 +5,9 @@ MKDIR=mkdir -p #hipcc HIPCC=/opt/rocm/bin/hipcc +#ROCm include directory +ROCM_INCLUDE=/opt/rocm/include + #BLAS ROCBLAS_LIB=rocblas @@ -21,15 +24,15 @@ OPT=-O3 conv: $(MKDIR) $(BIN_DIR) - $(HIPCC) ${SOURCE_DIR}/conv_bench_rocm.cpp -o $(BIN_DIR)/conv_bench -I$(DEEPBENCH_INC) -l$(MIOPEN_LIB) $(OPT) -std=c++11 --amdgpu-target=gfx900 + $(HIPCC) ${SOURCE_DIR}/conv_bench_rocm.cpp -o $(BIN_DIR)/conv_bench -I$(DEEPBENCH_INC) -I$(ROCM_INCLUDE) -l$(MIOPEN_LIB) $(OPT) -std=c++11 --amdgpu-target=gfx900 rnn: $(MKDIR) $(BIN_DIR) - $(HIPCC) ${SOURCE_DIR}/rnn_bench_rocm.cpp -o $(BIN_DIR)/rnn_bench -I$(DEEPBENCH_INC) -l$(MIOPEN_LIB) $(OPT) -std=c++11 --amdgpu-target=gfx900 + $(HIPCC) ${SOURCE_DIR}/rnn_bench_rocm.cpp -o $(BIN_DIR)/rnn_bench -I$(DEEPBENCH_INC) -I$(ROCM_INCLUDE) -l$(MIOPEN_LIB) $(OPT) -std=c++11 --amdgpu-target=gfx900 gemm: $(MKDIR) $(BIN_DIR) - $(HIPCC) ${SOURCE_DIR}/gemm_bench.cpp -o $(BIN_DIR)/gemm_bench -I$(DEEPBENCH_INC) -l$(ROCBLAS_LIB) $(OPT) -std=c++11 --amdgpu-target=gfx900 + $(HIPCC) ${SOURCE_DIR}/gemm_bench.cpp -o $(BIN_DIR)/gemm_bench -I$(DEEPBENCH_INC) -I$(ROCM_INCLUDE) -l$(ROCBLAS_LIB) $(OPT) -std=c++11 --amdgpu-target=gfx900 clean: rm -rf $(BIN_DIR)