#  cudadl version 0.9: Compute discrete logs in smooth group orders
#                      using CUDA
#  Copyright (C) 2012 by Ryan Henry and Ian Goldberg
#                        {rhenry,iang}@cs.uwaterloo.ca
#
#  This program is free software: you can redistribute it and/or modify
#  it under the terms of version 3 of the GNU General Public License as
#  published by the Free Software Foundation.
#
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License
#  along with this program.  If not, see <http://www.gnu.org/licenses/>.

-include Makefile.local

GMP ?= /usr/local
CUDA ?= /usr/local/cuda
LIBEVENT ?= /usr/local
MPI ?= /usr/local
CXX ?= g++

WORDS = 24

CXXFLAGS=-g -Wall -O2
CPPFLAGS=-DWORDS=$(WORDS) -I$(LIBEVENT)/include -I$(GMP)/include -UVERBOSE
NVCCOPTS=-g -arch sm_20 --ptxas-options=-v -O2
NVCC=nvcc $(NVCCOPTS)

CXXFILES = dlrho.cc gen_N.cc
CUFILES = parrhoasm.cu dpstream.cu
OFILES = dlrho.o gen_N.o cudadl.o controller.o controller_main.o evutils.o dpnode.o dpnode_main.o worker.o worker_main.o
NOMPI_TARGETS = gen_N dlrho controller dpnode worker
TARGETS = $(NOMPI_TARGETS) mpi

all: $(TARGETS)

nompi: $(NOMPI_TARGETS)

gen_N: gen_N.o
	g++ -g -Wall $^ -o $@ -lntl -L$(GMP)/lib -lgmp

dlrho: dlrho.o cudadl.o
	g++ -g -Wall $^ -o $@ -lntl -L$(GMP)/lib -lgmp -L$(CUDA)/lib64 -lcudart

dlrho.o: dlrho.cc
	g++ $(CXXFLAGS) $(CPPFLAGS) -I$(CUDA)/include $^ -c -o $@

cudadl.o: parrhoasm.cu cios.asm dpstream.cu
	$(NVCC) $(CPPFLAGS) -c parrhoasm.cu -DWORDS=$(WORDS) -o cudadl.o

cios.asm: gencios_reg_20
	./gencios_reg_20 $(WORDS) > $@

controller: controller.o evutils.o controller_main.o
	g++ -g -Wall $^ -o $@ -L$(LIBEVENT)/lib -Wl,-rpath=$(LIBEVENT)/lib -levent -lntl -L$(GMP) -lgmp

dpnode: dpnode.o evutils.o dpnode_main.o
	g++ -g -Wall $^ -o $@ -L$(LIBEVENT)/lib -Wl,-rpath=$(LIBEVENT)/lib -levent -lntl -L$(GMP) -lgmp

worker: worker.o evutils.o cudadl.o worker_main.o
	g++ -g -Wall $^ -o $@ -L$(LIBEVENT)/lib -Wl,-rpath=$(LIBEVENT)/lib -levent -levent_pthreads -lntl -L$(GMP) -lgmp -lpthread -L$(CUDA)/lib64 -lcudart

desres: controller_main.cc
	g++ $(CXXFLAGS) $(CPPFLAGS) -DTEST_DESIRED_RESOURCES $^ -o $@ -L$(LIBEVENT)/lib -Wl,-rpath=$(LIBEVENT)/lib -lntl -lgmp

mpi: mpi.o
	mpiCC -g -Wall $^ -o $@ -L/work/iang/sw/lib -lntl -L$(GMP) -lgmp -lpthread

mpi.o: mpi.cc
	mpiCC $(CXXFLAGS) $(CPPFLAGS) $^ -c -o $@

clean:
	-rm -f $(OFILES)

veryclean: clean
	-rm -f $(TARGETS) cios.asm