diff --git a/.gitignore b/.gitignore index 7dbf6e6..03f3c37 100644 --- a/.gitignore +++ b/.gitignore @@ -15,7 +15,6 @@ dist/ downloads/ eggs/ .eggs/ -lib/ lib64/ parts/ sdist/ diff --git a/lib/BUILD b/lib/BUILD index aafdfb4..3ddbc09 100644 --- a/lib/BUILD +++ b/lib/BUILD @@ -3,7 +3,7 @@ load("tf_custom_op_library") tf_custom_op_library( name = "deform_conv.so", - srcs = ["deform_update.cc", "deform_conv_util.h"], + srcs = ["deform_conv.cc", "deform_conv_util.h"], gpu_srcs = ["deform_conv.cu.cc", "deform_conv.h"], ) diff --git a/lib/deform_update.cc b/lib/deform_conv.cc similarity index 100% rename from lib/deform_update.cc rename to lib/deform_conv.cc diff --git a/lib/g++_complie.sh b/lib/g++_complie.sh index c62462d..e6b2bb4 100644 --- a/lib/g++_complie.sh +++ b/lib/g++_complie.sh @@ -1 +1 @@ -g++ -std=c++11 -shared -o deform_conv.so deform_update.cc deform_conv.cu.o -I $TF_INC -fPIC -lcudart -L $CUDA_HOME/lib64 -D GOOGLE_CUDA=1 -Wfatal-errors -I $CUDA_HOME/include -D_GLIBCXX_USE_CXX11_ABI=0 +g++ -std=c++11 -shared -o deform_conv.so deform_conv.cc deform_conv.cu.o -I $TF_INC -fPIC -lcudart -L $CUDA_HOME/lib64 -D GOOGLE_CUDA=1 -Wfatal-errors -I $CUDA_HOME/include -D_GLIBCXX_USE_CXX11_ABI=0