Skip to content

Commit 56015fe

Browse files
committed
refactor code
1 parent 62051d3 commit 56015fe

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

setup.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,14 +17,14 @@
1717
ext_modules=[
1818
CUDAExtension(
1919
'carafe_cuda',
20-
['src/carafe_cuda.cpp', 'src/carafe_cuda_kernel.cu'],
20+
['src/cuda/carafe_cuda.cpp', 'src/cuda/carafe_cuda_kernel.cu'],
2121
extra_compile_args={
2222
'cxx': [],
2323
'nvcc': NVCC_ARGS
2424
}),
2525
CUDAExtension(
2626
'carafe_naive_cuda',
27-
['src/carafe_naive_cuda.cpp', 'src/carafe_naive_cuda_kernel.cu'],
27+
['src/cuda/carafe_naive_cuda.cpp', 'src/cuda/carafe_naive_cuda_kernel.cu'],
2828
extra_compile_args={
2929
'cxx': [],
3030
'nvcc': NVCC_ARGS

0 commit comments

Comments
 (0)