Skip to content
Snippets Groups Projects
Commit b09afdf4 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Remove deprecation warning in test

parent b1098c24
No related branches found
No related tags found
No related merge requests found
......@@ -41,8 +41,8 @@ def test_jit():
from torch.utils.cpp_extension import CUDAExtension
lltm_cuda = CUDAExtension(
join(dirname(__file__), 'lltm_cuda'), [cpp_file, cuda_file], verbose=True, extra_cuda_cflags=[])
lltm_cuda = CUDAExtension(join(dirname(__file__), 'lltm_cuda'),
[cpp_file, cuda_file])
assert lltm_cuda is not None
print('hallo')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment