ciflow/trunk/199254: [FBGEMM] Fix cached BF16 embedding bags during typed half migration
- PyTorch 近 90 天出现 1986 次
- PyTorch 近 90 天第 1959 次发版
- 上一次：同一天稍早 · ciflow/trunk/199610: [T273452478] Cache FBGEMM embedding kernel handles (#199610)
发生了什么
Summary: Fix cached BF16 embedding-bag kernels selecting FP16 code under the pinned FBGEMM half-type aliases, while migrating downstream callers toward the typed half APIs. NOTE: No linked task. Please associate a task with this diff. Pinned FBGEMM currently aliases fbgemm::float16 and fbgemm::bfloat16 to uint16_t . ATen therefore keeps an explicit compile-time BF16 discriminator as a compatibility bridge, removes i…
摘要按规则整理自下方来源原文