setup.py.back
No OneTemporary
Actions

Subscribers

None

File Metadata

	from distutils.core import setup, Extension
	from torch.utils.cpp_extension import BuildExtension, CUDAExtension

	module1 = CUDAExtension('hdtorch', sources = ['hdtorch.cpp', 'hdtorch_cu.cu'],
	extra_compile_args=["-O3"])

	setup(name='hdtorch',
	ext_modules=[module1],
	cmdclass={'build_ext': BuildExtension})