Spaces:

yslan
/

GaussianAnything-AIGC3D

Running on Zero

GaussianAnything-AIGC3D / utils /emd /setup.py

init

7f51798 almost 2 years ago

296 Bytes

from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension

setup(
    name='emd',
    ext_modules=[
        CUDAExtension('emd', [
            'emd.cpp',
            'emd_cuda.cu',
        ]),
    ],
    cmdclass={
        'build_ext': BuildExtension
    })

	from setuptools import setup
	from torch.utils.cpp_extension import BuildExtension, CUDAExtension

	setup(
	name='emd',
	ext_modules=[
	CUDAExtension('emd', [
	'emd.cpp',
	'emd_cuda.cu',
	]),
	],
	cmdclass={
	'build_ext': BuildExtension
	})