opencompass icon indicating copy to clipboard operation
opencompass copied to clipboard

Published 20 hours ago • verified publisher icon open-compass

[Feature] 兼容torch_npu

Open li126com opened this issue 1 year ago • 1 comments

Describe the feature

目前看到代码中强依赖于torch.cuda，希望可以更改接口并兼容npu卡，即兼容torch_npu。

Will you implement it?

  • [ ] I would like to implement this feature and create a PR!

li126com avatar Apr 12 '24 08:04 li126com

Thanks for the great suggestions. We will add this feature into our backlog.

tonysy avatar Apr 15 '24 09:04 tonysy