viable/strict/1789084224: [ROCm] Enable UVM tests via hip-python's cuda.bindings interop (#196492)
- PyTorch 近 90 天出现 773 次
- PyTorch 近 90 天第 763 次发版
- 上一次：同一天稍早 · viable/strict/1789080651
发生了什么
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…
摘要按规则整理自下方来源原文