File size: 2,438 Bytes
da2e2ac |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
import os
import torch
from torch.utils.cpp_extension import (BuildExtension, CppExtension,
CUDAExtension)
import setuptools
# Change directory to allow installation from anywhere
script_folder = os.path.dirname(os.path.realpath(__file__))
os.chdir(script_folder)
with open("requirements.txt") as f:
requirements = f.read().splitlines()
def make_cuda_ext(name,
module,
sources,
sources_cuda=[],
extra_args=[],
extra_include_path=[]):
define_macros = []
extra_compile_args = {'cxx': [] + extra_args}
if torch.cuda.is_available() or os.getenv('FORCE_CUDA', '0') == '1':
define_macros += [('WITH_CUDA', None)]
extension = CUDAExtension
extra_compile_args['nvcc'] = extra_args + [
'-D__CUDA_NO_HALF_OPERATORS__',
'-D__CUDA_NO_HALF_CONVERSIONS__',
'-D__CUDA_NO_HALF2_OPERATORS__',
]
sources += sources_cuda
else:
print('Compiling {} without CUDA'.format(name))
extension = CppExtension
# raise EnvironmentError('CUDA is required to compile MMDetection!')
return extension(
name='{}.{}'.format(module, name),
sources=[os.path.join(*module.split('.'), p) for p in sources],
include_dirs=extra_include_path,
define_macros=define_macros,
extra_compile_args=extra_compile_args)
# Installs
setuptools.setup(
name="navsim",
version="1.0.0",
author="University of Tuebingen",
author_email="[email protected]",
description="TODO",
url="TODO",
python_requires=">=3.9",
packages=setuptools.find_packages(script_folder),
package_dir={"": "."},
classifiers=[
"Programming Language :: Python :: 3.9",
"Operating System :: OS Independent",
"License :: Free for non-commercial use",
],
license="apache-2.0",
install_requires=requirements,
ext_modules=[
make_cuda_ext(
name='bev_pool_v2_ext',
module='det_map.det.dal.mmdet3d.ops.bev_pool_v2',
sources=[
'src/bev_pool.cpp',
'src/bev_pool_cuda.cu',
],
),
],
cmdclass={'build_ext': BuildExtension},
)
|