viable/strict/1788040196: Fix static launcher 16-bit float scalars (#184065)
- PyTorch 近 90 天出现 239 次
- PyTorch 近 90 天第 236 次发版
- 上一次:同一天稍早 · viable/strict/1788035047
发生了什么
Triton reports fp16 and bf16 scalar arguments as distinct 16-bit types, but the static launcher mapped both to the 32-bit float ABI code. CUDA and HIP therefore wrote four-byte values into slots that kernels read as two-byte scalars, while XPU bound every imported-kernel argument as a pointer-sized value. Assign distinct fp16 and bf16 ABI codes and pack them with the same semantics as Triton's generated launchers: d…
摘要按规则整理自下方来源原文