wehub-resource-sync
24 lines
581 B
Python
24 lines
581 B
Python
#!/usr/bin/env python3
|
|
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
#
|
|
# This source code is licensed under the MIT license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
|
|
from setuptools import setup
|
|
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
|
|
|
|
|
|
setup(
|
|
name="lightconv_layer",
|
|
ext_modules=[
|
|
CUDAExtension(
|
|
"lightconv_cuda",
|
|
[
|
|
"lightconv_cuda.cpp",
|
|
"lightconv_cuda_kernel.cu",
|
|
],
|
|
),
|
|
],
|
|
cmdclass={"build_ext": BuildExtension},
|
|
)
|