Blame view

src/cudamatrix/Makefile 1.11 KB
8dcb6dfcb   Yannick Estève   first commit
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
  
  all:
  
  include ../kaldi.mk
  LDFLAGS += $(CUDA_LDFLAGS)
  LDLIBS += $(CUDA_LDLIBS)
  
  TESTFILES = cu-vector-test cu-matrix-test cu-math-test cu-test cu-sp-matrix-test cu-packed-matrix-test cu-tp-matrix-test \
              cu-block-matrix-test cu-matrix-speed-test cu-vector-speed-test cu-sp-matrix-speed-test cu-array-test \
  	    cu-sparse-matrix-test cu-device-test cu-rand-speed-test cu-compressed-matrix-test
  
  OBJFILES = cu-device.o cu-math.o cu-rand.o cu-matrix.o cu-packed-matrix.o cu-sp-matrix.o \
             cu-vector.o cu-common.o cu-tp-matrix.o cu-block-matrix.o \
             cu-sparse-matrix.o cu-allocator.o cu-array.o cu-compressed-matrix.o
  ifeq ($(CUDA), true)
    OBJFILES += cu-kernels.o
  endif
  
  LIBNAME = kaldi-cudamatrix
  
  ADDLIBS = ../util/kaldi-util.a ../matrix/kaldi-matrix.a ../base/kaldi-base.a 
  
  # Make sure we have CUDA_ARCH from kaldi.mk,
  ifeq ($(CUDA), true)
    ifndef CUDA_ARCH
      $(error CUDA_ARCH is undefined, run 'src/configure')
    endif
  endif
  
  # Implicit rule for kernel compilation,
  %.o : %.cu
  	$(CUDATKDIR)/bin/nvcc -c $< -o $@ $(CUDA_INCLUDE) $(CUDA_FLAGS) $(CUDA_ARCH) -I../
  
  include ../makefiles/default_rules.mk