Skip to content

GitLab

Commit 367b0af0 authored by rusty1s's avatar rusty1s
Browse files

c99 fix

parent f2897b93
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
build.py
View file @ 367b0af0
...@@ -13,6 +13,7 @@ sources = ['torch_scatter/src/cpu.c'] ...@@ -13,6 +13,7 @@ sources = ['torch_scatter/src/cpu.c']
include_dirs = ['torch_scatter/src'] include_dirs = ['torch_scatter/src']
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():
...@@ -33,6 +34,7 @@ ffi = create_extension( ...@@ -33,6 +34,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