지원하는 연산
Eager mode 실행 환경에서 기능적 정확성이 검증된 개별 연산 목록은 다음과 같습니다.
아래 연산들은 대부분의 경우 Rebellions NPU에서 가속되고 나머지 PyTorch의 연산들은 CPU로 실행됩니다. Rebellions NPU에서 가속되는 연산들은 점차 늘어날 예정입니다.
| PyTorch Ops | Note |
|---|---|
| torch.abs | - |
| torch.add | - |
| torch.addmm | - |
| torch.bmm | - |
| torch.ceil | - |
| torch.clamp | - |
| torch.div | - |
| torch.floor | - |
| torch.log | - |
| torch.masked_fill | - |
| torch.max | - |
| torch.mean | - |
| torch.min | - |
| torch.mm | - |
| torch.mul | - |
| torch.neg | - |
| torch.pow | - |
| torch.rsqrt | - |
| torch.silu | - |
| torch.softmax | - |
| torch.sub | - |
| torch.trunc | - |
| torch.where | - |