add-uint-support
Dodaj obsługę typów unsigned integer do operatorów PyTorch poprzez aktualizację makr AT_DISPATCH.
Instalacja
Wybierz klienta i sklonuj repozytorium do odpowiedniego katalogu skilli.
Instalacja
O skillu
Skill do PyTorch, który ułatwia dodanie obsługi typów bez znaku (uint16, uint32, uint64) do operatorów i kerneli. Automatyzuje proces aktualizacji makr AT_DISPATCH, umożliwiając rozszerzenie zakresu typów obsługiwanych przez istniejące operacje. Przydatny dla deweloperów pracujących nad rozszerzeniem funkcjonalności operatorów PyTorch o typy unsigned integer, szczególnie gdy trzeba włączyć kUInt16, kUInt32 lub kUInt64 w implementacjach kerneli.
Jak używać
Zidentyfikuj operator lub kernel, do którego chcesz dodać obsługę typów unsigned integer (uint16, uint32, uint64). Sprawdź, czy plik zawiera makra AT_DISPATCH.
Jeśli plik używa starego AT_DISPATCH (bez V2), najpierw skonwertuj go do AT_DISPATCH_V2, ponieważ skill pracuje z wersją V2 makr.
Zlokalizuj istniejące makro AT_DISPATCH_V2 w kodzie operatora. Zwykle wygląda ono jak: AT_DISPATCH_V2(dtype, "op", AT_WRAP(& { kernel
(); }), AT_EXPAND(AT_ALL_TYPES)). Rozszerz makro o obsługę typów unsigned, dodając AT_EXPAND(AT_BAREBONES_UNSIGNED_TYPES) jako dodatkowy argument, lub zastąp AT_ALL_TYPES na AT_INTEGRAL_TYPES_V2, jeśli pracujesz z typami całkowitymi.
Weryfikuj, że dodane typy (kUInt16, kUInt32, kUInt64) są obsługiwane przez kernel. Upewnij się, że implementacja kernela nie zawiera założeń specyficznych dla typów ze znakiem.
Przetestuj zmiany, aby potwierdzić, że operator poprawnie obsługuje nowe typy unsigned integer w różnych scenariuszach użycia.