Was macht china-ai-chip-skill?
China AI Chip Agent Skills für Metax und Ascend – spezialisierte Fähigkeiten für chinesische KI-Beschleuniger-Hardware.
Übersicht
Dieses Skill bietet KI-Agenten detaillierte Anleitungen zur Nutzung von Metax- und Ascend-KI-Chips. Es deckt folgende Bereiche ab:
- Modellportierung: Anpassung gängiger Modelle (z. B. ResNet, BERT) an die Hardwarearchitektur.
- Optimierung: Empfehlungen für Kernel-Tuning, Speicherverwaltung und Parallelisierung.
- Fehlerbehebung: Häufige Fehler bei der Kompilierung und Ausführung auf chinesischen KI-Chips.
- Leistungsanalyse: Messung von Durchsatz, Latenz und Energieeffizienz.
Unterstützte Plattformen
| Hersteller | Chip-Familie | SDK |
|---|---|---|
| Metax | M-Serie | MetaxSDK |
| Huawei | Ascend 910/310 | CANN |
Installation
-
Stellen Sie sicher, dass das entsprechende SDK (CANN oder MetaxSDK) installiert ist.
-
Fügen Sie dieses Skill zu Ihrem Agenten hinzu (z. B. via
~/.claude/skills). -
Konfigurieren Sie die Umgebungsvariablen:
export CHIP_TARGET=ascend # oder metax export CANN_PATH=/usr/local/Ascend
Beispiele
Modell auf Ascend ausführen
import torch
import torch_npu
model = torch.load('model.pth')
model.npu()
output = model(input_npu)
Metax-Optimierung
metax-opt --model resnet50.onnx --target mx500 --batch-size 32
Lizenz
MIT