ColossalAI icon indicating copy to clipboard operation
ColossalAI copied to clipboard

[BUG]: setup.py excludes op_builder

Open yynil opened this issue 2 years ago • 0 comments

🐛 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

yynil avatar Apr 08 '23 00:04 yynil