Agent skill · pytorch

cuda-index-width

Choose 32-bit vs 64-bit index math in PyTorch CUDA kernels. Use when fixing large-tensor indexing overflows, deciding whether to use int64_t, canUse32BitIn

Reference it in any coding agent with:

@skills pytorch/cuda-index-width

Browse the @skills marketplace