未解析的引用 'torch_mlu'

[appreciatehxy]

Prerequisite

• I have searched Issues and Discussions but cannot get the expected help.
• The bug has not been fixed in the latest version(https://github.com/open-mmlab/mmcv).

Environment

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]

What chip did you use?