Agent skill · nvidia
nvshmem-enable-tma
Prepare or review NVSHMEM CUDA kernels for TMA SMEM registration and direct-SMEM transfers. Do not use for unrelated CUDA tuning.
What it needs
About 5k tokens when loaded.
What this skill does
Enable TMA in NVSHMEM Applications Purpose Prepare an application for NVSHMEM's TMA-backed device-side transfers without changing its put/get APIs or unrelated kernels. Assess the target, choose the correct shared-memory path, make the smallest safe change, and explain launch-time requirements. Prerequisites Require readable CUDA/C++ source for a patch or code review. A static assessment needs no GPU, installed NVSHMEM runtime, or cluster access. Require an NVSHMEM 3.7.0-or-newer development environment and CUDA toolchain only when compiling. Require SM90-or-newer GPUs and peer GPU load/store reachability such as NVLink only when validating TMA execution or performance. CFT requires TMA. Reference Routing Before changing code, read registration-and-launch.md completely and then every reference selected below. Use the references as shapes, not blind replacements: preserve the application's error handling, launch abstraction, completion contract, and shared-memory layout. Situation Required additional reference --- --- Global-memory operands, or staging copy/buffer is not proven redundant gmem-staging.md Explicitly opted-in shared-memory put source with a proven-redundant GMEM copy direct-smem-put.md Compile, run, diagnose, or report validation validation-and-troubleshooting.md For a read-only assessment or review, read only the references needed to evaluate the proposed path. Do not make a code change until the complete registration reference and every selected path reference have been read. Inputs Required: Application source or source path, and the requested outcome: assessment, patch, review, or debugging. Optional: Derive NVSHMEM version, CUDA architecture, topology, kernel launch sites, and launch configuration from the project or environment when possible. Conditional for execution: Obtain build command, launch command, and target hardware access only when the user asks to compile or run validation. …
How to use it
Reference it in AdaL, Claude Code, Cursor or any coding agent — nothing to install:
@skills nvidia/nvshmem-enable-tma