setup.py

from setuptools import setup, find_packages
from torch.utils.cpp_extension import BuildExtension, CppExtension, CUDAExtension
import os

def load_extensions():
    return [CppExtension(
        name=os.path.join('test_gammagl', 'ceshi', 'ext', '_ext').replace(os.path.sep, "."),
        sources=[
            'test_gammagl/ceshi/ext/first.cpp'
        ]
    )]


classifiers = [
    'Development Status :: 3 - Alpha',
    'License :: OSI Approved :: Apache Software License',
]

setup(
    name="test_gammagl",
    version="0.1",
    author="zhou guangyu",
    author_email="1029175863@qq.com",
    maintainer="zhou guangyu",
    license="Apache-2.0 License",
    url="https://github.com/gyzhou2000/setup",
    download_url="https://github.com/gyzhou2000/setup",
    python_requires='>=3.6',
    cmdclass={'build_ext': BuildExtension},
    ext_modules=load_extensions(),
    packages=find_packages(),
    classifiers=classifiers
)