Commit 988e3595 authored by rusty1s's avatar rusty1s
Browse files

added files

parents
__pycache__/
_ext/
build/
dist/
.cache/
.eggs/
.coverage
*.egg-info/
*.so
# PyTorch Spline Convolution
from torch.utils.ffi import create_extension
headers = ['torch_spline_conv/src/cpu.h']
sources = ['torch_spline_conv/src/cpu.c']
include_dirs = ['torch_spline_conv/src']
define_macros = []
extra_objects = []
with_cuda = False
ffi = create_extension(
name='torch_spline_conv._ext.ffi',
package=True,
headers=headers,
sources=sources,
include_dirs=include_dirs,
define_macros=define_macros,
extra_objects=extra_objects,
with_cuda=with_cuda,
relative_to=__file__)
if __name__ == '__main__':
ffi.build()
[aliases]
test=pytest
[tool:pytest]
addopts = --capture=no --cov
from os import path as osp
from setuptools import setup, find_packages
import build # noqa
install_requires = ['cffi']
setup_requires = ['pytest-runner', 'cffi']
tests_require = ['pytest', 'pytest-cov']
setup(
name='torch_spline_conv',
version='0.1.0',
description='PyTorch extension for spline-based convolutions',
url='https://github.com/rusty1s/pytorch_spline_conv',
author='Matthias Fey',
author_email='matthias.fey@tu-dortmund.de',
install_requires=install_requires,
setup_requires=setup_requires,
tests_require=tests_require,
packages=find_packages(exclude=['build']),
ext_package='',
cffi_modules=[osp.join(osp.dirname(__file__), 'build.py:ffi')],
)
import torch_spline_conv as SplineConv
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