Full AlexNet Fwd+Bwd using MiOpen+hipblas

patflick · Jul 11, 2017 · 3188b82 · 3188b82
1 parent a4cec26
commit 3188b82
Show file tree

Hide file tree

Showing 2 changed files with 452 additions and 126 deletions.
diff --git a/Makefile b/Makefile
@@ -2,7 +2,7 @@ ROCM_PATH?= $(wildcard /opt/rocm)
 HIP_PATH?= $(wildcard /opt/rocm/hip)
 HIPCC=$(HIP_PATH)/bin/hipcc
 INCLUDE_DIRS=-I$(HIP_PATH)/include -I$(ROCM_PATH)/include
-LD_FLAGS=-L$(ROCM_PATH)/lib -L$(ROCM_PATH)/opencl/lib/x86_64 -lMIOpen -lOpenCL
+LD_FLAGS=-L$(ROCM_PATH)/lib -L$(ROCM_PATH)/opencl/lib/x86_64 -lMIOpen -lOpenCL -lhipblas-hcc -lrocblas-hcc -lprofiler
 TARGET=--amdgpu-target=gfx900
 
 HIPCC_FLAGS=-g $(CXXFLAGS) $(TARGET) $(INCLUDE_DIRS)