import os
from setuptools import setup, Extension
from torch.utils import cpp_extension

os.environ['CPPFLAGS'] = '-I/home/herrmalu/packages/pytorch_sparse/csrc'
os.environ['CPPFLAGS'] = '-I/home/herrmalu/packages/cpp/tqdm.cpp/include'
os.environ['LDFLAGS'] = '-L/home/herrmalu/packages/pytorch_sparse/build'
# setup(name='o2conv1d_cpp',
#       ext_modules=[cpp_extension.CppExtension(name='o2conv1d_cpp', 
#                                            sources=['o2conv1d.cpp'],
#                                            extra_compile_args=[],
#                                            extra_objects=['-o libtorchsparse.so']
#                                            )],
#       cmdclass={'build_ext': cpp_extension.BuildExtension})
setup(name='hessian_cpp',
      ext_modules=[cpp_extension.CppExtension(name='hessian_cpp', 
                                           sources=['hessian.cpp'],
                                           extra_compile_args=[],
                                           extra_objects=['-o libtorchsparse.so']
                                           )],
      cmdclass={'build_ext': cpp_extension.BuildExtension})