Merge pull request #89 from ROCmSoftwarePlatform/master

AMD GPU support for conv, gemm, rnn for fp32.

README.md

@@ -844,3 +844,49 @@ To compile and run the benchmark, please use the following command:
 make sparse EIGEN_PATH=<path>
 bin/sparse_bench
 ```
+
+# AMD Benchmarks
+
+## Prerequisites
+* A ROCm enabled platform, more info [here](https://rocm.github.io/install.html).
+* [MIOpen](https://github.com/ROCmSoftwarePlatform/MIOpen) - HIP backend of MIOpen is required.
+* [rocBLAS](https://github.com/ROCmSoftwarePlatform/rocBLAS)
+
+At present only `fp32 train` benchmarks are enabled.
+
+## Compiling
+
+The `Makefile` in `code/amd` is for an AMD `gfx900` GPU. To benchmark other generations, please modify the `Makefile` accordingly.
+
+Setting your enviroment variables before compiling/running:
+
+```
+export PATH=PATH_TO_ROCM/bin:$PATH
+export CPATH=PATH_TO_MIOPEN/include:$CPATH
+export LIBRARY_PATH=PATH_TO_MIOPEN/lib:$LIBRARY_PATH
+export LD_LIBRARY_PATH=PATH_TO_MIOPEN/lib:PATH_TO_MIOPENGEMM/lib:$LD_LIBRARY_PATH
+```
+
+To compile the convolution, RNNs and GEMM benchmarks, run:
+
+```
+make conv rnn gemm
+```
+
+## Running the Benchmarks
+After successful compilation, the executables will be generated in the `bin` directory.
+
+To benchmark convolutions:
+```
+bin/conv_bench
+```
+
+To benchmark RNN:
+```
+bin/rnn_bench
+```
+
+To benchmark GEMM:
+```
+bin/gemm_bench
+```

code/amd/Makefile

@@ -0,0 +1,35 @@
+SOURCE_DIR?=.
+BIN_DIR?=bin
+MKDIR=mkdir -p
+
+#hipcc
+HIPCC=/opt/rocm/bin/hipcc
+
+#BLAS
+ROCBLAS_LIB=rocblas
+
+#CONV
+MIOPEN_LIB?=MIOpen
+
+#DeepBench
+DEEPBENCH_INC=${SOURCE_DIR}/../kernels
+
+all: conv rnn gemm 
+
+#OPT=-g -O0 -fsanitize=undefined -fno-omit-frame-pointer
+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
+
+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
+
+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
+
+clean:
+	rm -rf $(BIN_DIR)