[BUG] fix bug in Carafe upsample
@@ -11,15 +11,10 @@ setup(
name='carafe',
ext_modules=[
CUDAExtension(
- 'carafe_cuda',
- ['src/carafe_cuda.cpp', 'src/carafe_cuda_kernel.cu'],
- extra_compile_args={
- 'cxx': [],
- 'nvcc': NVCC_ARGS
- }),
- CUDAExtension(
- 'carafe_naive_cuda',
- ['src/carafe_naive_cuda.cpp', 'src/carafe_naive_cuda_kernel.cu'],
+ 'carafe_ext', [
+ 'src/cuda/carafe_cuda.cpp', 'src/cuda/carafe_cuda_kernel.cu',
+ 'src/carafe_ext.cpp'
+ ],
extra_compile_args={
'cxx': [],
'nvcc': NVCC_ARGS
