fix setup bug

Author: leftthomas

README.md

@@ -21,7 +21,7 @@ pip install --upgrade git+https://github.com/leftthomas/CARAFE.git@master
 
 ## Example
 ```python
-from carafe.carafe import CARAFE, CARAFENaive
+from carafe import CARAFE, CARAFENaive
 module = CARAFE(kernel_size=5, group_size=1, scale_factor=4)
 module = CARAFENaive(kernel_size=5, group_size=1, scale_factor=4)
 ```

setup.py

@@ -16,19 +16,21 @@
     license='MIT',
     ext_modules=[
         CUDAExtension(
-            'carafe_cuda',
-            ['src/cuda/carafe_cuda.cpp', 'src/cuda/carafe_cuda_kernel.cu'],
+            'carafe_ext',
+            ['src/carafe_ext.cpp', 'src/carafe_cuda.cpp', 'src/carafe_cuda_kernel.cu'],
             extra_compile_args={
                 'cxx': [],
                 'nvcc': NVCC_ARGS
             }),
         CUDAExtension(
-            'carafe_naive_cuda',
-            ['src/cuda/carafe_naive_cuda.cpp', 'src/cuda/carafe_naive_cuda_kernel.cu'],
+            'carafe_naive_ext',
+            ['src/carafe_naive_ext.cpp', 'src/carafe_naive_cuda.cpp', 'src/carafe_naive_cuda_kernel.cu'],
             extra_compile_args={
                 'cxx': [],
                 'nvcc': NVCC_ARGS
             })
     ],
+    package_data={'.': ['*.so']},
     cmdclass={'build_ext': BuildExtension},
-    install_requires=['torch'])
+    zip_safe=False,
+    install_requires=['torch>=1.5'])