ciflow/trunk/196920: [ROCm] Keep the capture handle's pointer mode per stream
- PyTorch 近 90 天出现 1612 次
- PyTorch 近 90 天第 1593 次发版
- 上一次:同一天稍早 · viable/strict/1790639441: [OSDC] Remove map_ec2_to_arc.py (#198706)
发生了什么
getCaptureCublasHandle serves every capturing stream of a thread with one rocBLAS handle, but it copied the pointer mode from the stream's eager handle only the first time a (capture, handle, stream) was seen. A caller that set device pointer mode on stream A, then used stream B (host mode) in the same capture, got B's host mode back on A, so device-pointer scalars were read as host pointers. Each capture workspace…
摘要按规则整理自下方来源原文