Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "6b1abba18dce2eb169b6363ea5f626e7cd87cf21"
Commit 3fc02192 authored by rusty1s's avatar rusty1s
Browse files

rename

parent 70249ef7
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
setup.py
View file @ 3fc02192
...@@ -14,7 +14,7 @@ cmdclass = {} ...@@ -14,7 +14,7 @@ cmdclass = {}
if torch.cuda.is_available(): if torch.cuda.is_available():
ext_modules += [ ext_modules += [
CUDAExtension('matmul_cuda', CUDAExtension('matmul_cuda',
['cuda/matmul.cpp', 'cuda/matmul_cuda.cu']) ['cuda/matmul.cpp', 'cuda/matmul_kernel.cu'])
] ]
cmdclass['build_ext'] = BuildExtension cmdclass['build_ext'] = BuildExtension
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment