open-mmlab / mmcv

OpenMMLab Computer Vision Foundation
https://mmcv.readthedocs.io/en/latest/
Apache License 2.0
5.91k stars 1.65k forks source link

未解析的引用 'torch_mlu' #3068

Open appreciatehxy opened 8 months ago

appreciatehxy commented 8 months ago

Prerequisite

Environment

bug

Reproduces the problem - code sample

在 'utils.py' 中找不到引用 'build_extension'

try: import torch if torch.version == 'parrots': from parrots.utils.build_extension import BuildExtension EXT_TYPE = 'parrots' elif (hasattr(torch, 'is_mlu_available') and torch.is_mlu_available()) or \ os.getenv('FORCE_MLU', '0') == '1': from torch_mlu.utils.cpp_extension import BuildExtension EXT_TYPE = 'pytorch' else: from torch.utils.cpp_extension import BuildExtension EXT_TYPE = 'pytorch' cmd_class = {'build_ext': BuildExtension}

Reproduces the problem - command or script

.

Reproduces the problem - error message

.

Additional information

No response

zhouzaida commented 7 months ago

What chip did you use?