Khanin/merged pool emb opt#4977

Closed

kudomcho wants to merge 2 commits intopytorch:mainfrom

ROCm:khanin/merged_pool_emb_opt

Commits on Oct 1, 2025

fixed the correctness of merged pool embedding
yxsamliu
committed

Commits on Oct 7, 2025

ufmt formatted, added use_pitch on given at unit test, and added pitch calculation for better memory alignment on merged_embeddings_benchmark.py
yxsamliu
committed