Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
S
seminar-breakout
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Analytics
Analytics
CI / CD
Repository
Value Stream
Wiki
Wiki
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Shashank Suhas
seminar-breakout
Commits
14ea8af3
Commit
14ea8af3
authored
Apr 06, 2017
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
add ycm_conf. and better build
parent
bde13a8d
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
29 additions
and
8 deletions
+29
-8
tensorpack/user_ops/.ycm_extra_conf.py
tensorpack/user_ops/.ycm_extra_conf.py
+23
-0
tensorpack/user_ops/Makefile
tensorpack/user_ops/Makefile
+5
-7
tensorpack/user_ops/__init__.py
tensorpack/user_ops/__init__.py
+1
-1
No files found.
tensorpack/user_ops/.ycm_extra_conf.py
0 → 100644
View file @
14ea8af3
import
tensorflow
as
tf
flags
=
[
'-Wall'
,
'-Wextra'
,
'-Werror'
,
'-Wno-long-long'
,
'-Wno-variadic-macros'
,
'-fexceptions'
,
'-std=c++11'
,
'-x'
,
'c++'
,
'-isystem'
,
tf
.
sysconfig
.
get_include
()
]
def
FlagsForFile
(
filename
,
**
kwargs
):
return
{
'flags'
:
flags
,
'do_cache'
:
True
}
tensorpack/user_ops/Makefile
View file @
14ea8af3
...
...
@@ -9,19 +9,17 @@ OPTFLAGS ?= -O3 -march=native
#OPTFLAGS ?= -g3 -fsanitize=address,undefined -O2 -lasan
#OPTFLAGS ?= -g3 -fsanitize=leak -O2 -lubsan
#
optional extra packages
#LIBS = opencv
#
INCLUDE_DIR += $(shell pkg-config --cflags $(LIBS))
#
LDFLAGS += $(shell pkg-config $(LIBS) --libs)
#
extra packages from pkg-config
LIBS
=
protobuf libzmq
INCLUDE_DIR
+=
$(
shell
pkg-config
--cflags
$(LIBS)
)
LDFLAGS
+=
$(
shell
pkg-config
$(LIBS)
--libs
)
CXXFLAGS
?=
CXXFLAGS
+=
$(INCLUDE_DIR)
CXXFLAGS
+=
-Wall
-Wextra
-Wno-unused-parameter
-Wno-ignored-qualifiers
CXXFLAGS
+=
-Wall
-Wextra
CXXFLAGS
+=
$(DEFINES)
-std
=
c++11
$(OPTFLAGS)
-fPIC
CXXFLAGS
+=
-D_GLIBCXX_USE_CXX11_ABI
=
0
# https://github.com/tensorflow/tensorflow/issues/1569
LDFLAGS
+=
$(OPTFLAGS)
LDFLAGS
+=
-lzmq
-lprotobuf
$(OMP_FLAG)
LDFLAGS
+=
-shared
-fPIC
SHELL
=
bash
...
...
tensorpack/user_ops/__init__.py
View file @
14ea8af3
...
...
@@ -11,7 +11,7 @@ __all__ = ['zmq_recv']
include_dir
=
tf
.
sysconfig
.
get_include
()
file_dir
=
os
.
path
.
dirname
(
os
.
path
.
abspath
(
__file__
))
compile_cmd
=
'make INCLUDE_DIR=
-I{} -C {}
'
.
format
(
include_dir
,
file_dir
)
compile_cmd
=
'make INCLUDE_DIR=
"-isystem {}" -C "{}"
'
.
format
(
include_dir
,
file_dir
)
print
(
"Compiling user ops ..."
)
ret
=
os
.
system
(
compile_cmd
)
if
ret
!=
0
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment