"references/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "742fd13c17f3e61390fd28924c71f92f18c7efe3"
Commit 3fc02192 authored by rusty1s's avatar rusty1s
Browse files

rename

parent 70249ef7
......@@ -14,7 +14,7 @@ cmdclass = {}
if torch.cuda.is_available():
ext_modules += [
CUDAExtension('matmul_cuda',
['cuda/matmul.cpp', 'cuda/matmul_cuda.cu'])
['cuda/matmul.cpp', 'cuda/matmul_kernel.cu'])
]
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