Skip to content

GitLab

Commit b87eab0b authored by rusty1s's avatar rusty1s
Browse files

c99 fix

parent d37028f5
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
build.py
View file @ b87eab0b
...@@ -15,6 +15,7 @@ sources = ['aten/TH/TH{}.c'.format(f) for f in files] ...@@ -15,6 +15,7 @@ sources = ['aten/TH/TH{}.c'.format(f) for f in files]
include_dirs = ['aten/TH'] include_dirs = ['aten/TH']
define_macros = [] define_macros = []
extra_objects = [] extra_objects = []
extra_compile_args = ['-std=c99']
with_cuda = False with_cuda = False
if torch.cuda.is_available(): if torch.cuda.is_available():
...@@ -35,6 +36,7 @@ ffi = create_extension( ...@@ -35,6 +36,7 @@ ffi = create_extension(
include_dirs=include_dirs, include_dirs=include_dirs,
define_macros=define_macros, define_macros=define_macros,
extra_objects=extra_objects, extra_objects=extra_objects,
extra_compile_args=extra_compile_args,
with_cuda=with_cuda, with_cuda=with_cuda,
relative_to=__file__) relative_to=__file__)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment