ciflow/trunk/199254: [FBGEMM] Fix cached BF16 embedding bags during typed half migration
- PyTorch: 1981 events in the last 90 days
- PyTorch: 1954th Release in the last 90 days
- Previous: earlier the same day · ciflow/trunk/199610: [T273452478] Cache FBGEMM embedding kernel handles (#199610)
What happened
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…
Summary assembled by rule from the sources below