Open appreciatehxy opened 8 months ago
在 '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}
.
No response
What chip did you use?
Prerequisite
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