ColossalAI
ColossalAI copied to clipboard
[BUG]: setup.py excludes op_builder
🐛 Describe the bug
The setup.py in main branch just excludes op_builders.
setup(name=package_name,
version=version,
packages=find_packages(exclude=(
'op_builder',
'benchmark',
'docker',
'tests',
'docs',
'examples',
'tests',
'scripts',
'requirements',
'*.egg-info',
)),
I'm not sure why others may get colossalai work without op_builder. In my machine using the latest main trunk, I must copy op_builder directory to site_packages directory to get it work even after I pre-build the ops using CUDA_EXT=1 flag.
Environment
Cuda: 11.7 Python: 3.10 OS: Ubuntu 22.04 Pytorch:1.13with cuda 11.7