← 返回事件
持续讨论AI发版

ciflow/trunk/196920: [ROCm] Keep the capture handle's pointer mode per stream

发生了什么

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…

摘要按规则整理自下方来源原文

为什么在扩散

来源