MUSAExtension扩展包导入失败

[gtgtgt1117]

参考示例脚本替换为自己得mu和cpp文件，

python导入时报错如下： 包版本如下： setuptools 68.2.2 torch 2.0.0 torch-musa 1.1.0 torchaudio 2.0.1+3b40834 torchvision 0.15.2a0+fa99a53

[zehan-mt]

请这样使用，

import torch
import torch_musa
from freqencoder import FreqEncoder
view_encoder = FreqEncoder(input_dim=3, degree=4)
dirs_musa = torch.tensor([[1.1,2.2,3.3],[4.4,5.5,6.6]], requires_grad=True, device="musa")

def compute(dirs):
    dirs_ = dirs/dirs.norm(dim=-1,keepdim=True)
    code = view_encoder(dirs_)
    loss = code.mean()
    loss.backward()
    return code

print(compute(dirs_musa), dirs_musa.grad)