← Back to events
ActiveTechRelease7.15

viable/strict/1789084224: [ROCm] Enable UVM tests via hip-python's cuda.bindings interop (#196492)

Photo: PyTorch Releases

What happened

torch.cuda._use_uvm() already works on ROCm: it imports cuda.bindings.runtime directly, and hip-python's interop package (PyPI: hip-python-interop) provides a HIP-backed cuda.bindings with the cuda-python 12+ module layout, verified live against HIP 7.15 (cudaMallocManaged / cudaMemAdvise / cudaFree behave with the (err, *outs) convention, and its error enums compare equal across the cuda/hip enum classes so cuda_py…

Summary assembled by rule from the sources below

Why it's spreading

Sources