Include parlai/clib/cuda/ngram_repeat_block_cuda_kernel.*
Include parlai/clib/cuda/ngram_repeat_block_cuda_kernel.* files, otherwise I can't seem to build models. For example,
parlai.core.agents.create_agent_from_mode_file('zoo:bb3/bb3_3B/model')
fails with
No such file or directory: ../clib/cuda/ngram_repeat_block_cuda.cpp
@@ -37,7 +37,7 @@ if __name__ == '__main__':
packages=find_packages(exclude=('data', 'docs', 'tests', 'parlai_internal*')),
install_requires=reqs,
include_package_data=True,
- package_data={'': ['*.txt', '*.md', '*.opt']},
+ package_data={'': ['*.txt', '*.md', '*.opt', '*.cu', '*.cpp']},
entry_points={
"flake8.extension": ["PAI = parlai.utils.flake8:ParlAIChecker"],
"console_scripts": ["parlai=parlai.__main__:main"],
