Skip to content

Commit 427f869

Browse files
committed
remove unnecessary lib and target path. switch to use /usr/local/cuda
1 parent d40c589 commit 427f869

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

Makefile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,10 +37,10 @@ all: $(TARGET_LIB)
3737
$(NVCC) -c -o $@ $< $(CFLAGS) -I. -DGOOGLE_CUDA=1 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86 -x cu -Xcompiler -fPIC --expt-relaxed-constexpr
3838

3939
%_kernels.cc.o: distributed_embeddings/cc/kernels/%_kernels.cc distributed_embeddings/cc/kernels/%.h
40-
$(CXX) -c -o $@ $< $(CFLAGS) -Wall -fPIC -I/usr/local/cuda/targets/x86_64-linux/include
40+
$(CXX) -c -o $@ $< $(CFLAGS) -Wall -fPIC -I/usr/local/cuda/include
4141

4242
$(TARGET_LIB): $(NVCC_OBJS) $(CXX_OBJS) distributed_embeddings/cc/ops/embedding_lookup_ops.cc
43-
$(CXX) $(CFLAGS) -fPIC -o $@ $^ $(LDFLAGS) -lnvToolsExt -L/usr/local/cuda/targets/x86_64-linux/lib
43+
$(CXX) $(CFLAGS) -fPIC -o $@ $^ $(LDFLAGS) -L/usr/local/cuda/lib64
4444

4545
pip_pkg: $(TARGET_LIB)
4646
bash build_pip_pkg.sh

third_party/gpus/cuda/include

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
/usr/local/cuda/targets/x86_64-linux/include
1+
/usr/local/cuda/include

0 commit comments

Comments
 (0)