Skip to content

GitLab

Commit 81940250 authored by rusty1s's avatar rusty1s
Browse files

reset

parent 2f389cce
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
build.py
View file @ 81940250
...@@ -11,9 +11,7 @@ if osp.exists('build'): ...@@ -11,9 +11,7 @@ if osp.exists('build'):
files = ['serial', 'grid'] files = ['serial', 'grid']
headers = ['torch_cluster/src/{}_cpu.h'.format(f) for f in files] headers = ['torch_cluster/src/{}_cpu.h'.format(f) for f in files]
headers += ['aten/TH/THGreedy.h', 'aten/TH/THGrid.h']
sources = ['torch_cluster/src/{}_cpu.c'.format(f) for f in files] sources = ['torch_cluster/src/{}_cpu.c'.format(f) for f in files]
sources += ['aten/TH/THGreedy.c', 'aten/TH/THGrid.c']
include_dirs = ['torch_cluster/src', 'aten/TH'] include_dirs = ['torch_cluster/src', 'aten/TH']
define_macros = [] define_macros = []
extra_objects = [] extra_objects = []
......
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