tensor-bindings-gpu

Installation
SKILL.md

Tensor Bindings: GPU Read and Write

GPU tensor bindings use GPU-mode PhysX with DLPack CUDA tensors. DirectGPU (/physics/suppressReadback) is a separate setting: keep it enabled for fastest tensor-pipeline workloads and disabled for workflows that need contact modification. ovstage populates the authored scene, while binding path resolution also includes runtime-only clone paths. The tensor data path does not go through ovstage.

When to Use

Use this skill when a caller needs GPU-to-GPU tensor exchange through CUDA device pointers, PyTorch CUDA tensors, or DLPack without CPU staging.

Instructions

  1. Read the full C or Python sample before adapting this pattern because CUDA memory lifetime, DLPack shape storage, and device ordinal handling matter.
  2. Select an explicit CUDA ordinal with active_cuda_gpus, allocate memory on that device before wrapping it in DLTensor, and keep shape storage valid until the synchronous read or write returns.
  3. Use Shell to compile and run the full sample or a local integration test after adapting the scene path and tensor type.
Installs
1
GitHub Stars
4.7K
First Seen
7 days ago
tensor-bindings-gpu — nvidia-omniverse/physx