[FEA]: NVSHMEM Support
@aterrel y travaille déjà.
Depuis le 10/12/2025.
Évaluation
Cette issue n'a pas encore été évaluée.
Description
Is this a new feature, an improvement, or a change to existing functionality?
New Feature
How would you describe the priority of this feature request?
Critical (currently preventing usage)
Please provide a clear description of problem this feature solves
Is NVSHMEM integration planned for cuTile? Lack of NVSHMEM support prevents kernels from performing fine-grained, in-kernel communication, limiting compute–communication overlap.
Feature Description
Support to NVSHMEM device APIs (e.g. nvshmemx_putmem_block)
Describe your ideal solution
@ct.kernel
def vector_add(a, b, remote_c, tile_size: ct.Constant[int], pe: ct.Constant[int]):
# Get the 1D pid
pid = ct.bid(0)
# Load input tiles
a_tile = ct.load(a, index=(pid,), shape=(tile_size,))
b_tile = ct.load(b, index=(pid,), shape=(tile_size,))
# Perform elementwise addition
result = a_tile + b_tile
# Store result using NVSHMEM
ct. nvshmemx_putmem_block(remote_c, result, index=(pid,), pe=pe)
Describe any alternatives you have considered
No response
Additional context
No response
Contributing Guidelines
- I agree to follow cuTile Python's contributing guidelines
- I have searched the open feature requests and have found no duplicates for this feature request
- Langage dominant
- Python
- Étoiles
- 2.2k
- Forks
- 155
- Métriques de merge des PR
- Aucune PR mergée en 30 j
Préparer son environnement
- Aucun Dockerfile ni fichier Docker Compose
- Propose un modèle de pull request
- Lire le guide de contribution
Par où commencer
- Lisez l'issue en entier, puis le guide de contribution du projet.
- Signalez en commentaire que vous la prenez — cela évite que deux personnes fassent le même travail.
- Forkez le dépôt et travaillez sur une branche.
- Ouvrez une pull request qui référence le numéro de l'issue.
Autres issues de NVIDIA/cutile-python
-
[BUG]: check_dtype_support rejects family-conditional (sm_XXXa) gpu_code targetsPeut-être pris @sylvesterkaczmarek l’a pris il y a 32 jours. Ouverte
Difficulté 2/5 1-3 heures Accessibilité débutants 82/100
NVIDIA/cutile-python#105 · 2 commentaires ·
-
nvidia-runners
Difficulté 2/5 1-3 heures Accessibilité débutants 25/100
NVIDIA/cutile-python#108 ·
-
[BUG]: FFT sample launches `Batch` blocks that each process the full batchPeut-être pris @AntonOresten l’a pris il y a 140 jours. Ouvertebug status: needs-triage
Difficulté 3/5 1-2 jours Accessibilité débutants 68/100
NVIDIA/cutile-python#102 ·
-
Difficulté 4/5 3-5 jours Accessibilité débutants 68/100
NVIDIA/cutile-python#101 ·
-
bug
Difficulté 4/5 3-5 jours Accessibilité débutants 45/100
NVIDIA/cutile-python#97 · 1 commentaire ·
Toutes les issues de NVIDIA/cutile-python
Issues similaires
-
bug
Difficulté 1/5 Moins d'une heure Accessibilité débutants 88/100
openwisp/openwisp-utils#777 ·
-
Boolean statistics (Covered, Confident, JaccardDistant) aggregate to 1.0 without float weightsOuverte
Difficulté 2/5 1-3 heures Accessibilité débutants 63/100
google-research/weatherbenchX#310 ·
Les mainteneurs répondent en général sous 3 jours
-
Difficulté 2/5 1-3 heures Accessibilité débutants 78/100
huggingface/peft#3955 ·
Les mainteneurs répondent en général sous 1 jour
-
author-page correction
Difficulté 2/5 1-3 heures Accessibilité débutants 60/100
acl-org/acl-anthology#10313 ·
Les mainteneurs répondent en général sous 1 jour
-
bug
Difficulté 2/5 1-3 heures Accessibilité débutants 76/100
AOSSIE-Org/Gitcord-GithubDiscordBot#147 · 1 commentaire ·
Les mainteneurs répondent en général sous 1 jour