Пользовательское расширение C ++ / CUDA

Question

Я следовал инструкциям здесь и создал собственное расширение CUDA для проекта PyTorch. Файл установки показан здесь.

from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension

setup(
    name='lltm',
    ext_modules=[
        CUDAExtension('lltm_cuda', [
            'lltm_cuda.cpp',
            'lltm_cuda_kernel.cu',
        ])
    ],
    cmdclass={
        'build_ext': BuildExtension
    })

Однако выполнение python setup.py build приводит к следующей ошибке:

error: parameter packs not expanded with '...'.

Ошибка в файле PyTorch. То есть это не имеет ничего общего с кодом, который я написал.

/opt/conda/lib/python3.6/site-packages/torch/include/c10/util/variant.h

Любая помощь / подсказка высоко ценится.