#  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
MACROS=-DDERANDOMIZE -UVERBOSE -DSAVE_DPS -DMAKE_VERSIONS_COMPARABLE

CXXFLAGS=-g -Wall -O2
CPPFLAGS=-DWORDS=$(WORDS) -I$(LIBEVENT)/include -I$(GMP)/include -std=c++11 $(MACROS)
NVCCOPTS=-g -arch sm_30 --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
PTX_TARGETS = parrhoasm.ptx
NOMPI_TARGETS = gen_N dlrho controller dpnode worker desres
TARGETS = $(NOMPI_TARGETS) mpi

all: $(TARGETS)

nompi: $(NOMPI_TARGETS)

ptx: $(PTX_TARGETS)

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

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

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

parrhoasm.ptx: parrhoasm.cu cios.asm dpstream.cu
	$(NVCC) $(CPPFLAGS) -c $< -o $@ --ptx

cudadl.o: parrhoasm.cu cios.asm dpstream.cu
	$(NVCC) $(CPPFLAGS) -c $< -o $@

test_cuda: parrhoasm.cu cios.asm dpstream.cu
	$(NVCC) -DTEST_CUDA $(CPPFLAGS) $< -o $@ -lntl -L$(GMP)/lib -lgmp -L$(CUDA)/lib64 -lcudart -lpthread

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

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

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

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

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

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

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) $(PTX_TARGETS)

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