Allow more flexible CUDA build environment

IDEA-Research / detrex

detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.

https://detrex.readthedocs.io/en/latest/

Apache License 2.0

1.99k stars 206 forks source link

Allow more flexible CUDA build environment #345

Closed rodriguhe closed 5 months ago

rodriguhe commented 5 months ago

This change allows for a correct build using get_extensions() to occur outside of a CUDA environment. This addresses the import error of import detrex._C when running multi_scale_deform_attn.py

rodriguhe commented 5 months ago

For example: https://github.com/facebookresearch/detectron2/blob/main/setup.py#L66-L68