NVIDIA/Fuser

Support tensor `rng_seed` and `rng_offset`

Open

#855 ouverte le 8 sept. 2023

Voir sur GitHub
 (0 commentaires) (0 réactions) (0 assignés)C++ (80 forks)github user discovery
Priority = P0good first issue

Métriques du dépôt

Stars
 (392 stars)
Métriques de merge PR
 (Métriques PR en attente)

Description

Currently only integer RNG states are supported as per https://github.com/NVIDIA/Fuser/blob/e3a0b3ef528a86012d0a5268f3065b9203ed2784/csrc/python_frontend/python_bindings.cpp#L2426-L2473.

I'd like to feature request the support of 0-dim Tensor as rng_seed and rng_offset as well. The dtype would be int64 and the device would be either cpu and cuda.

On top of that, I'd like the function to return a Tensor of integer or an integer that should be used to advance / increment the offset for the next function call.

Guide contributeur

Direction de recherche
Implémentez la prise en charge des tenseurs 0 dimensionnels en tant que rng seed et rng offset dans les liaisons Python, gérez les périphériques CPU et CUDA avec le dtype int64 et modifiez le retour pour produire un tenseur ou un entier pour l'avancement du décalage. Examinez l'implémentation RNG existante basée sur des entiers dans csrc/python frontend/python bindings.cpp et étendez le code pertinent.
Stack technique
cpppython
Domaine
backend
Type d'issue
Fonctionnalité
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
3
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
3-5 jours
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
C++Python
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
40